Lutz Roeder
6bcbbe1005
Add PyTorch test file
|
6 years ago | |
|---|---|---|
| .github | 6 years ago | |
| .vscode | 7 years ago | |
| setup | 6 years ago | |
| src | 6 years ago | |
| test | 6 years ago | |
| tools | 6 years ago | |
| .eslintrc.json | 6 years ago | |
| .gitattributes | 8 years ago | |
| .gitignore | 6 years ago | |
| DEVELOPMENT.md | 6 years ago | |
| LICENSE | 9 years ago | |
| Makefile | 6 years ago | |
| README.md | 6 years ago | |
| electron-builder.yml | 6 years ago | |
| package.json | 6 years ago | |
| setup.cfg | 6 years ago | |
| setup.py | 6 years ago |
README.md
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), Caffe (.caffemodel, .prototxt), Caffe2 (predict_net.pb, predict_net.pbtxt), Darknet (.cfg), MXNet (.model, -symbol.json), ncnn (.param) and TensorFlow Lite (.tflite).
Netron has experimental support for TorchScript (.pt, .pth), PyTorch (.pt, .pth), Torch (.t7), Arm NN (.armnn), BigDL (.bigdl, .model), Chainer (.npz, .h5), CNTK (.model, .cntk), Deeplearning4j (.zip), MediaPipe (.pbtxt), ML.NET (.zip), MNN (.mnn), OpenVINO (.xml), PaddlePaddle (.zip, __model__), scikit-learn (.pkl), TensorFlow.js (model.json, .pb) and TensorFlow (.pb, .meta, .pbtxt, .ckpt, .index).
Install
macOS: Download the .dmg file or run brew cask install netron
Linux: Download the .AppImage file or run snap install netron
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]').
Models
Sample model files to download or open using the browser version:
- ONNX: squeezenet [open]
- CoreML: exermote [open]
- Darknet: yolo [open]
- Keras: mobilenet [open]
- MXNet: inception_v3 [open]
- TensorFlow: chessbot [open]
- TensorFlow Lite: hair_segmentation [open]
- TorchScript: traced_online_pred_layer [open]
- Caffe: mobilenet_v2 [open]