本文主要是介绍学习经验分享【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可视化特征图教程(持续更新)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!