Netron
Netron 是一款深度神经网络可视化工具,这样一款神器的开发作者是微软的大神 Lutz Roeder ,并且在自己的家中完成的。
Netron 强大的原因在于:
- 所支持的平台广泛。不像 tensorboard 等较为“专一”的可视化平台,当前主流的深度学习框架,Netron 都能得到很好的支持;
- 操作简单快捷。不需要写一行代码,只需要下载软件安装,然后打开需要可视化的文件,一步操作即可,当然也可以通过代码实现;
- 保存快捷。对于可视化的结果,就像保存普通的文件一样,一步到位,保存在自己的电脑上。
一、支持的框架
Netron 最为强大的功能,就在于它所支持的框架十分广泛:
- ONNX:
.onnx
、.pb
、,pbtxt
- Keras:
.h5
、.keras
- CoreML:
.mlmodel
- Caffe2:
predict_net.pb
、predict_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. 在线使用
这种方式最为简单,只需要打开网页,放入模型就可以浏览了。
2. python 安装
打开终端,输入:
1
pip install netron
进入 python 解释器或新建一个 python 脚本,输入:
1
2
import netron
netron.start(r"<path/to/model>")
运行该代码会跳出一个服务网址,并自动打开浏览器,得到如下的效果:
这里提供一个 onnx 模型(链接: https://pan.baidu.com/s/1C2abik9L9e0XRJVhPIJFZA 提取码: 17ka),供读者体验效果。
如果觉得本教程不错或对您有用,请前往项目地址 https://github.com/Harry-hhj/Harry-hhj.github.io 点击 Star :) ,这将是对我的肯定和鼓励,谢谢!
三、参考文献
作者:Harry-hhj,github主页:传送门