Netron is a viewer for neural network, deep learning and machine learning models.
Netron supports ONNX (.onnx, .pb, .pbtxt), Keras (.h5, .keras), Core ML (.mlmodel), Caffe2 (predict_net.pb, predict_net.pbtxt), MXNet (.model, -symbol.json) and TensorFlow Lite (.tflite). Netron has experimental support for Caffe (.caffemodel, .prototxt), PyTorch (.pt, .pth), TorchScript (.pt, .pth), Torch (.t7), CNTK (.model, .cntk), PaddlePaddle (__model__), Darknet (.cfg), scikit-learn (.pkl), TensorFlow.js (model.json, .pb) and TensorFlow (.pb, .meta, .pbtxt).
Install
macOS: Download the .dmg file or run brew cask install netron
Linux: Download the .AppImage or .deb file.
Windows: Download the .exe installer.
Browser: Start the browser version.
Python Server: Run pip install netron and netron [FILE] or import netron; netron.start('[FILE]').
Download Models
Sample model files to download and open:
- ONNX: resnet-18
- Keras: tiny-yolo-voc
- CoreML: faces_model
- TensorFlow Lite: smartreply
- MXNet: inception_v1
- Caffe: mobilenet_v2
- TensorFlow: inception_v3