关键词:
遥感影像
小目标检测
YOLOv5s
多尺度特征融合
SimAM
摘要:
针对现有目标检测模型难以准确检测小目标并存在大量误检与漏检的问题,该文提出了一种基于YOLOv5s的遥感影像小目标检测算法。首先,在主干网络加入SimAM无参数注意力机制模块,在不增加额外参数的情况下,使算法更加关注重要特征,同时抑制遥感影像中背景信息;然后,采用SPD-Conv模块进行下采样,避免了特征信息的丢失;之后,引入了四尺度检测特征融合网络,以获得更丰富的多尺度特征信息,并优化融合网络和输出检测头结构。最终实现对于遥感影像小目标的精准检测。在自制的LEVIR-4SC遥感影像小目标数据集上的实验结果表明,与YOLOv5s相比,提出算法的模型参数减少了6.3%,精确度P和平均精度均值mAP达到了90.7%和83.7%,提升了6.5%和4.2%。与YOLOv8等经典方法相比参数量更少,模型更轻量化,在精确度、平均精度方面均有提升,证明提出算法具有更好的检测效果。