摘要
小目标检测作为目标检测任务中的一种,存在检测率低、特征信息不足和漏检率较高等问题。针对这些问题,本文提出一种改进YOLOv5的小目标检测算法,将聚集-分发(Gather-and-Distribute,GD)机制加入YOLOv5网络,该机制通过改进卷积和自注意力2种操作提高了多尺度融合能力,用于捕获不同尺度间像素级的关系,取得了延迟和准确性之间的理想平衡。改进后的算法在自制的砂眼数据集上进行试验,平均检测精度mAP指标达到98.8%,与基线模型相比,mAP值提高了2.5%,验证了改进算法的可行性。
出处
《中国新技术新产品》
2024年第7期21-24,共4页
New Technology & New Products of China