学习经验分享【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，即可得到相应的特征图