×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

【转载】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.

版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭