当前位置:首页 > 编程笔记 > 正文
已解决

学习经验分享【NO.18】YOLOv5可视化特征图教程(持续更新)

来自网友在路上 169869提问 提问时间:2023-11-05 10:02:52阅读次数: 69

最佳答案 问答题库698位专家为你答疑解惑

YOLOv5项目的6.0以上版本中的detect.pt中集成了可视化相关模块,直接调用即可。

一、可视化特征提取网络中所有模块的可视化图

添加形参如下所示,加载相应的权值文件后,选择相应的图片。

运行detect.py文件后得到如下所示:

 

 以stage0_Conv_features为例展示:

二、可视化特征提取网络中指定模块的可视化图(32通道)

需要在yolo.py文件中将改为如下,即可得到conv模块后的可视化特征图

运行detect.py可以得到如下所示: 

 

三、可视化特征提取网络中指定模块的合并通道可视化

在Plot.py中添加如下代码

在yolo.py中进行调用

运行detect.py,即可得到相应的特征图

 

查看全文

99%的人还看了

猜你感兴趣

版权申明

本文"学习经验分享【NO.18】YOLOv5可视化特征图教程(持续更新)":http://eshow365.cn/6-32618-0.html 内容来自互联网,请自行判断内容的正确性。如有侵权请联系我们,立即删除!