模型结构可视化神器——Netron(支持tf, caffe, keras,mxnet，onnx等多种框架)转化成onnx显示网络结构图

模型结构可视化神器——Netron(支持tf, caffe, keras,mxnet等多种框架)

原创 木盏 最后发布于2018-08-23 14:37:54 阅读数 26921
发布于2018-08-23 14:37:54
分类专栏： tools
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文处处链接和本声明。
本文链接：https://blog.csdn.net/leviopku/article/details/81980249

很多时候，复现人家工程的时候，需要了解人家的网络结构。但不同框架之间可视化网络层方法不一样，这样给研究人员造成了很大的困扰。
前段时间，发现了一个可视化模型结构的神奇：Netron

---

目前的Netron支持主流各种框架的模型结构可视化工作，我直接给出gayhub链接：
https://github.com/lutzroeder/Netron
支持windows，Linux，mac系统
在windows系统，下载一个.exe文件就很稳了，如下：

安装以后，就是只需双击打开，添加模型文件的位置就可以了。
Netron 支持的框架和对应文件如下：

框架	对应文件
ONNX	.onnx, .pb
Keras	.h5, .keras
CoreML	.mlmodel
TensorFlow Lite	.tflite

试验性支持，可能不同稳定：

框架	对应文件
Caffe	.caffemodel
Caffe2	predict_net.pb
MXNet	.model, -symbol.json
TensorFlow.js	model.json, .pb
TensorFlow	.pb, .meta

只需按照上述表格将对应的文件路径加到netron中，就可以看到漂亮的网络结构图了，截取部分显示：