jennyb2911 / netron

Visualizer for 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), CoreML (.mlmodel), Caffe2 (predict_net.pb, predict_net.pbtxt), MXNet (.model, -symbol.json) and TensorFlow Lite (.tflite). Netron has experimental support for Caffe (.caffemodel, .prototxt), 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 [MODEL_FILE]. In Python run import netron and netron.browse('model.onnx').

Download Models

Sample model files you can download and open:

ONNX Models: Inception v1, Inception v2, ResNet-50, SqueezeNet

Keras Models: resnet, tiny-yolo-voc

CoreML Models: MobileNet, Places205-GoogLeNet, Inception v3

TensorFlow Lite Models: Smart Reply 1.0 , Inception v3 2016

Caffe Models: BVLC AlexNet, BVLC CaffeNet, BVLC GoogleNet

Caffe2 Models: BVLC GoogleNet, Inception v2

MXNet Models: CaffeNet, SqueezeNet v1.1

TensorFlow models: Inception v3, Inception v4, Inception 5h

About

Visualizer for deep learning and machine learning models

https://www.lutzroeder.com/ai

License:MIT License


Languages

Language:JavaScript 97.5%Language:Python 1.5%Language:Shell 0.4%Language:HTML 0.4%Language:CSS 0.2%Language:Makefile 0.1%