Yolov5系列（一）——网络可视化工具netron

一、介绍

netron：网络可视化工具，使用这个软件可以清晰的展示网络模型的每一层的输入输出，网络的整体的架构，而且支持不同的网络框架。
类似下图：

二、安装步骤（介绍windows系统安装）

（其他版本可以参考博客）

1、打开此链接：Github下载链接

打开上述链接，划到下面 install，选择 windows 下载

2、选择下图所示Netron-Setup-5.8.9.exe，下载安装

3、安装成功之后显示如下：

三、打开yolov5系列的模型

1、转换

因为netron对 .pt 的文件没有太好的兼容性，所以需要将文件进行转换，在yolov5 工程中，提供了export.py，可以进行文件转换

2、在终端直接输入以下命令

python export.py --weights yolov5s.pt --include onnx

注意：输入命令时需要注意export.py 和 yolo5s.pt 的路径
我是直接在该文件的目录下，所以不需要再写其他的路径

执行完命令，会在yolov5s.pt 同级目录下生成 yolov5s.onnx
此时就可以用网络可视化工具netron打开

四、执行命令可能遇到的问题

requirements: onnx not found and is required by YOLOv5, attempting auto-update...
requirements: 'pip install onnx' skipped (offline)
ONNX: export failure: No module named 'onnx'

解决办法

安装onnx
pip install onnx

之后再执行上述命令就没有问题了