【netron】神经网络可视化

文章目录

  • 1 为什么选择netron
  • 2 如何安装
    • 2.1 命令行安装
    • 2.2 安装包本地安装
  • 3 如何使用
    • 3.1 命令行安装时使用方式
    • 3.2 安装包安装后使用方式

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,如下图,然后复制网址到浏览器即可

【netron】神经网络可视化_第1张图片

展示:

【netron】神经网络可视化_第2张图片

3.2 安装包安装后使用方式

安装完netron.exe后,onnx等文件打开方式一般会变成如下图所示,不是的话,打开方式选择netron即可。
【netron】神经网络可视化_第3张图片

双击打开效果如下图:
【netron】神经网络可视化_第4张图片

你可能感兴趣的:(深度学习基础知识,深度学习,神经网络)