netron安装使用

netron几种常见打开方法

1.netron github链接
2.netron 网页版
3.(python2.7)pip安装netron
pip install netron -i https://pypi.tuna.tsinghua.edu.cn/simple
(仅针对python2.7) 安装完毕后,需要pip show netron找到netron安装位置,修改netron/server.py文件。将import urllib.parse修改为import urlparse ,否则会报错ImportError: No module named parse.

4.在网页版netron无法加载时,方法3也无法正常打开netron。

使用

pip安装版使用时,命令行输入netron,单击打开出现的http://localhost:8080/链接,将模型拖入网页即可。用完后,命令行ctrl +c 。
网页版使用时,点击‘openmodel…’键,选择需要打开的模型,点击‘open’键确定即可。

netron支持的框架

ONNX (.onnx, .pb, .pbtxt), Keras (.h5, .keras), Core ML (.mlmodel), Caffe (.caffemodel, .prototxt), Caffe2 (predict_net.pb), Darknet (.cfg), MXNet (.model, -symbol.json), Barracuda (.nn), ncnn (.param), Tengine (.tmfile), TNN (.tnnproto), UFF (.uff) and TensorFlow Lite (.tflite).
有一些尚在试验中的框架,见官方介绍

特别说明,netron对于支持的框架,层与层之间有清晰的连接线,但对于尚在试验中的框架,可能会出现打开后只有单层信息,没有层间连线的情况。比如,pytorch。此时,建议将这些模型先转化为onnx模型,然后再用netron打开onnx模型。

你可能感兴趣的:(Python)