Bootstrap

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

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

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

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

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

 

 以stage0_Conv_features为例展示:

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

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

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

 

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

在Plot.py中添加如下代码

在yolo.py中进行调用

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

 

;