Netron 可视化工具对 yolov5 模型观测_netorn

一、Netron可视化工具的下载

Netron的下载地址：https://github.com/lutzroeder/netron/releases/tag/v5.3.3

打开链接之后，滑倒最底端，window 操作系统下载如下的文件，点击后安装即可 。

二、将 yolov5 模型中权重文件转化为 onnx 格式

如果只用yolov5中的权重文件（比如：yolov5s.pt 格式）直接在 Netron 打开，那么得到的网络结构图十分简单，参考价值不大。可以根据如下的步骤将 .pt 文件转化问 .onnx 格式。

首先找到yolov5模型中models文件夹下的export.py文件

打开之后唯一要修改的参数只有主函数中的权重文件的位置，找到自己下载的 yolov5 模型中权重文件的位置，复制路径粘贴到如下的位置，然后运行。

 

可能遇到的问题就是会提示有如下的错误:No mudule named 'onnx'，导致不能转化为onnx格式。

 只需要在虚拟环境下输入如下的指令：

pip install onnx

之后再运行，就可以看到在 models 模块中生成了 .onnx 文件。

三、在Netorn中打开  .onnx 文件

点击 Open Module

找到之前创建的 .onnx 文件，打开就看到对应的网络结构啦！