当前位置:网站首页>模型结构可视化神器——Netron(支持tf, caffe, keras,mxnet等多种框架)

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

2020-12-02 10:57:02 osc_8hwmnuba

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

版权声明
本文为[osc_8hwmnuba]所创,转载请带上原文链接,感谢
https://my.oschina.net/u/4409965/blog/4769751