Netron-模型结构可视化工具（支持tf, caffe, keras,mxnet,onnx,darknet）

Netron支持主流各种框架的模型结构可视化工作，github地址：https://github.com/lutzroeder/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).

支持macos,linux,windows安装包安装。支持浏览器和python server的web显示。

1、浏览器打开

最方便的为浏览器打开，浏览器地址：https://netron.app/。将模型文件拖入浏览器页面即可显示。如将yolov3.cfg文件拖入浏览器，可看到yolov3模型结构如下：

2、python server方式

step1、执行pip install netron安装包

step2、代码实现：

import netron

netron.start("model.onnx")

默认会在浏览器http://localhost:8080地址打开。若要修改端口，添加参数address=端口号即可。