1 为什么选择netron
在复现别人的模型的时候,有时我们要知道一个模型的输入与输出名,可是有时作者并没有告诉我们,要我们自己去查。使用netron
,可以清晰地看见网络的输入输出名,具体的网络结构。
支持的框架广泛,包括onnx、MXNet、TensorFlow、PyTorch等。
2 如何安装
2.1 命令行安装
激活虚拟环境,输入命令,对应使用方式见3.1节。
pip install netron
2.2 安装包本地安装
更多平台的安装包可见netron官网,担心科学上网不太方便,这里也提供windows版本的netron.exe
百度网盘的链接,使用方式见3.2节。
链接:https://pan.baidu.com/s/12KeZ2LXRAmT6EkKenkEOYg
提取码:fx4g
3 如何使用
3.1 命令行安装时使用方式
以onnx模型为例,输入命令netron onnx_model_name.onnx
,如下图,然后复制网址到浏览器即可
展示:
3.2 安装包安装后使用方式
安装完netron.exe后,onnx等文件打开方式一般会变成如下图所示,不是的话,打开方式选择netron即可。
双击打开效果如下图: