【转载】NETRON神经网络模型可视化工具 qwer2020-03-04 17:36:48 回复 1 查看 综合话题
【转载】NETRON神经网络模型可视化工具

项目地址:https://github.com/lutzroeder/netron



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.pbpredict_net.pbtxt), Darknet (.cfg), MXNet (.model-symbol.json), ncnn (.param) and TensorFlow Lite (.tflite).

Netron has experimental support for Torch  (.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 (. ), PaddlePaddle (.zip__model__), scikit-learn (.pkl), TensorFlow.js (model.json.pb) and TensorFlow (.pb. .pbtxt.ckpt.index).

Install

macOSDownload the .dmg file or run brew cask install netron

LinuxDownload the .AppImage file or run snap install netron

WindowsDownload the .exe installer.

BrowserStart 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:



  • © 2020 GitHub, Inc.

热门帖子

精华帖子

官方微博 官方微信
版权所有 © 2019 寒武纪 Cambricon 备案/许可证号:京ICP备17003415
关闭