一、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 文件，打开就看到对应的网络结构啦！
Original: https://blog.csdn.net/weixin_46088823/article/details/121144440
Author: Jay_Mapp
Title: Netron 可视化工具对 yolov5 模型观测
原创文章受到原创版权保护。转载请注明出处：https://www.johngo689.com/686502/
转载文章受原作者版权保护。转载请注明原作者出处！