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=端口号即可。