DocF / netron

Visualizer for neural network, deep learning and machine learning models

Home Page:https://www.lutzroeder.com/ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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), MXNet (.model, -symbol.json), TorchScript (.pt, .pth), NCNN (.param) and TensorFlow Lite (.tflite).

Netron has experimental support for PyTorch (.pt, .pth), Torch (.t7), CNTK (.model, .cntk), Deeplearning4j (.zip, configuration.json), PaddlePaddle (.zip, __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:

About

Visualizer for neural network, deep learning and machine learning models

https://www.lutzroeder.com/ai

License:MIT License


Languages

Language:JavaScript 94.9%Language:Python 2.5%Language:Shell 1.0%Language:CSS 0.8%Language:HTML 0.6%Language:Makefile 0.1%