This Visual Studio Code extension uses Netron that allows users to load and visualize machine learning and deep learning models, generated from the following frameworks and toolkits: ONNX, PyTorch, TensorFlow, TensorFlow Lite, OpenVINO, PaddlePaddle, MXNet etc.
Currently the extension supports these file formats: *.pt, *.pth, *.pb, *.mlmodel, *.tflite, *.onnx, *.h5, *.keras, *.paddle, *.pickle, *.safetensors, *.xmodel and many more !
Feel free to test the extension with models downloaded from the ONNX Model Zoo or 🤗 Hugging Face.
You can also open the Netron web app by selecting Start Netron web on your Command Palette
(Only work if you are using the extension on your host machine or with WSL).
Special thanks to Lutz Roeder and all the developers who contributed to the development of Netron.