神经网络模型可视化工具netron

1、支持的框架

support for：
ONNX (.onnx, .pb, .pbtxt),
Keras (.h5, .keras),
CoreML (.mlmodel),
Caffe2 (predict_net.pb, predict_net.pbtxt),
MXNet (.model, -symbol.json)
TensorFlow Lite (.tflite).
experimental support for ：
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).

2、安装

支持linux、windows、mac
查看：https://github.com/lutzroeder/Netron
工具里面的说明进行安装，推荐用pip安装，很方便

pip install netron

如果不想安装的话也没问题，作者提供了一个在线view的网页，只要上传模型就能生成结构了
https://lutzroeder.github.io/netron/

3、使用

可以建立一个python文件，如下内容，将模型地址名称写入即可，运行会出现一个网页，即可显示自己的模型结构以及其名称

import netron
modelPath = "googlenet.pb"
netron.start(modelPath)