模型可视化工具netron

netron:

在这里插入图片描述

详情可见官方github:https://github.com/lutzroeder/netron

使用方法

为了方便使用,我们直接用它的网页版,进入网址:https://netron.app/,直接点击接受,打开模型即可。
在这里插入图片描述
在这里插入图片描述
例如打开一个.pb模型(左上角为菜单选项,点击模型相应位置,右侧会显示具体网络参数):
aGV_FFFFFF,t_70)

yolov5中的使用

.pt模型

虽然可以打开.pt模型,但模型的细节显示不出来。
在这里插入图片描述

模型转换

利用yolov5中的模型转化代码将.pt模型转为onnx格式。
打开yolov5-master/models/export.py文件,改好自己模型的参数,直接运行即可。
在这里插入图片描述
在运行上述脚本前可以选择在环境中pip install onnx也可以不装,不装的话转换出的模型为:
在这里插入图片描述
可视化效果:
在这里插入图片描述
安装onnx
在这里插入图片描述

运行export.py转换出的为:
在这里插入图片描述
可视化效果:
在这里插入图片描述

Guess you like

Origin blog.csdn.net/weixin_48994268/article/details/119669541