平衡锚框位置偏差的大核注意力小目标检测方法

分享到

【获取途径】超星期刊网
【作者】冉同霄，闫建红
【刊名】太原师范学院学报(自然科学版)
【作者单位】太原师范学院计算机科学与技术学院
【年份】2024
【卷号】第23卷
【期号】第4期
【页码】10-18
【ISSN】1672-2027
【关键词】YOLOv5 小目标检测 Wise IoU 大型可分离核注意力归一化Wasserstein距离
【摘要】针对小目标图像像素低、背景复杂和特征难提取等问题，提出改进的YOLOv5小目标检测模型.首先，为扩大感受野，增强模型多尺度特征提取能力，基于CSP思想，在YOLOv5s模型的SPPF模块中，引入相似性注意力和大型可分离核注意力，组成大核注意力的金字塔池化方法；其次，为解决锚框质量和位置偏差的问题，使用能聚焦普通质量锚框的WIoU损失函数，在此基础上，增加归一化Wasserstein距离NWD指标，即WIoU-NWD损失函数；最后，为进一步增强检测头表示能力，获取更有用的小目标特征，在检测头使用融...
【文献类型】期刊