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

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

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

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

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

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

 以stage0_Conv_features为例展示:学习经验分享【NO.18】YOLOv5可视化特征图教程(持续更新)_第2张图片

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

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

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

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

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

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

在Plot.py中添加如下代码

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

在yolo.py中进行调用

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

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

你可能感兴趣的:(学习经验分享,YOLO算法改进系列,学习,YOLO,人工智能,计算机视觉)