nkamsteve / 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), Keras (.h5, .keras), CoreML (.mlmodel) and TensorFlow Lite (.tflite). Netron has experimental support for Caffe (.caffemodel), Caffe2 (predict_net.pb), MXNet (-symbol.json), TensorFlow.js (model.json, .pb) and TensorFlow (.pb, .meta).

Install

macOS

Download the .dmg file or with Homebrew 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].
Serve a model in Python using import netron; netron.serve_file('my_model.onnx').

Download Models

Sample model files you can download and open:

ONNX Models

Keras Models

CoreML Models

TensorFlow Lite Models

Caffe Models

Caffe2 Models

MXNet Models

TensorFlow models

About

Visualizer for deep learning and machine learning models

https://www.lutzroeder.com/ai

License:MIT License


Languages

Language:JavaScript 98.5%Language:Python 0.9%Language:Shell 0.3%Language:HTML 0.2%Language:CSS 0.1%Language:Makefile 0.0%