Home Netron
Post
Cancel

Netron

Preview Image

Netron

Netron 是一款深度神经网络可视化工具,这样一款神器的开发作者是微软的大神 Lutz Roeder ,并且在自己的家中完成的。

Netron 强大的原因在于:

  1. 所支持的平台广泛。不像 tensorboard 等较为“专一”的可视化平台,当前主流的深度学习框架,Netron 都能得到很好的支持;
  2. 操作简单快捷。不需要写一行代码,只需要下载软件安装,然后打开需要可视化的文件,一步操作即可,当然也可以通过代码实现;
  3. 保存快捷。对于可视化的结果,就像保存普通的文件一样,一步到位,保存在自己的电脑上。

一、支持的框架

Netron 最为强大的功能,就在于它所支持的框架十分广泛:

  • ONNX: .onnx.pb,pbtxt
  • Keras: .h5.keras
  • CoreML: .mlmodel
  • Caffe2: predict_net.pbpredict_net.pbtxt
  • MXNet: .model-symbol.json
  • Tensorflow Lite: tflite
  • Caffe: .caffemodel.prototxt
  • PyTorch: .pth
  • Torch: .t7
  • CNTK: .model.cntk
  • PaddlePaddle: __model__
  • Darknet: cfg
  • scikit-learn: .pkl
  • TensorFlow.js: .model.json.pb
  • TensorFlow: .pb.meta.pbtxt

二、安装

1. 在线使用

这种方式最为简单,只需要打开网页,放入模型就可以浏览了。

image-20210904205820798

2. python 安装

打开终端,输入:

1
pip install netron

进入 python 解释器或新建一个 python 脚本,输入:

1
2
import netron
netron.start(r"<path/to/model>")

运行该代码会跳出一个服务网址,并自动打开浏览器,得到如下的效果:

example.onnx

这里提供一个 onnx 模型(链接: https://pan.baidu.com/s/1C2abik9L9e0XRJVhPIJFZA 提取码: 17ka),供读者体验效果。


如果觉得本教程不错或对您有用,请前往项目地址 https://github.com/Harry-hhj/Harry-hhj.github.io 点击 Star :) ,这将是对我的肯定和鼓励,谢谢!


三、参考文献

  1. 【神经网络可视化01】——用Netron实现可视化

作者:Harry-hhj,github主页:传送门

This post is licensed under CC BY 4.0 by the author.