Loovelj / 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), 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), Darknet (.cfg), ML.NET (.zip), MNN (.mnn), OpenVINO (.xml), PaddlePaddle (.zip, __model__), 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 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]').

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 96.1%Language:Python 1.9%Language:Shell 0.9%Language:CSS 0.5%Language:HTML 0.5%Language:Makefile 0.1%