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

转自：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中，就可以看到漂亮的网络结构图了，截取部分显示： 

--------------------- 
作者：木盏 
来源：CSDN 
原文：https://blog.csdn.net/leviopku/article/details/81980249 
版权声明：本文为博主原创文章，转载请附上博文链接！