关键词:
河湖“四乱”
目标检测
多样化分支块
动态检测头
损失函数
摘要:
针对河湖“四乱”遥感图像中目标检测精度低、小目标检测难等问题,提出一种基于改进YOLOv5的河湖“四乱”目标检测算法。首先,以YOLOv5s为基础,采用多样化分支块代替原有的C3模块,以增强主干网络的特征提取能力。其次,引入具有统一注意力机制的动态检测头,以增强算法的目标检测和定位能力。最后,设计新的损失函数ShapeIoU,降低目标尺度差异带来的影响,以提升模型的泛化能力。将该算法在自建的河湖“四乱”数据集上进行实验,结果表明,模型的检测精度为60.6%、召回率为55.7%、mAP50为56.1%、AP50∶95为32.2%,较基准网络分别提升了5.3%、5.3%、6.7%和4.6%。此外,在公开数据集RSOD和NWPU VHR-10上的实验表明,该算法与几种经典算法相比,整体均表现出较大优势。