特征图教程（持续更新）"/>
学习经验分享【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，即可得到相应的特征图
更多推荐
学习经验分享【NO.18】YOLOv5可视化特征图教程（持续更新）
发布评论