首页|改进YOLOv7遥感图像目标检测算法

改进YOLOv7遥感图像目标检测算法

扫码查看
针对遥感图像中目标尺度变化大和背景复杂导致检测精度低的问题,设计了一种改进YOLOv7目标检测算法.首先,为了缓解复杂背景对检测器的干扰,设计了一个注意力引导的高效层聚合网络(ALAN),以优化多路径网络使其更聚焦前景目标而降低背景的影响;其次,为了降低目标尺度变化大对检测精度的影响,设计了一种注意力多尺度特征增强(AMSFE)模块,用于扩展主干网络输出特征的感受野,以加强网络对尺度变化大目标的特征表征能力;最后,引入旋转边界框损失函数,以获取任意朝向物体的精确位置信息.在DIOR-R数据集上的实验结果表明,该算法mAP达到了 64.51%,相比于基线原始YOLOv7 算法提高了 3.43%,且优于其他同类算法,能够适应遥感图像中多尺度和复杂背景的目标检测任务.
Improving YOLOv7 remote sensing image target detection algorithm
In response to the problem of low detection accuracy caused by significant object scale variations and complex backgrounds in remote sensing images,an improved YOLOv7 object detection algorithm was designed.Firstly,in order to alleviate the interference of complex backgrounds on the detector,an attention-guided efficient layer aggregation network(ALAN)was designed to optimize the multi-path network to focus more on foreground objects,thereby reducing the impact of background.Secondly,in order to reduce the impact of significant object scale variations on detection accuracy,an attention multi-scale feature enhancement(AMSFE)module was designed to expand the receptive field of the backbone network output features,enhancing the network's feature representation ability for objects with substantial scale variations.Finally,a rotating bounding box loss function was introduced to obtain precise location information of objects in any orientation.The experimental results on the DIOR-R dataset demonstrated that the proposed algorithm achieved a mean average precision(mAP)of 64.51%,an improvement of 3.43%over the baseline original YOLOv7 algorithm.Furthermore,it outperformd other similar algorithms and was capable of handling object detection tasks in remote sensing images with multi-scale and complex backgrounds.

remote sensingobject detectionfeature enhancementattention mechanismYOLOv7

李大湘、吉展、刘颖、唐垚

展开 >

西安邮电大学通信与信息工程学院,陕西 西安 710121

遥感 目标检测 特征增强 注意力机制 YOLOv7

国家自然科学基金项目陕西省自然科学基金项目

620713792019JM-604

2024

图学学报
中国图学学会

图学学报

CSTPCD北大核心
影响因子:0.73
ISSN:2095-302X
年,卷(期):2024.45(4)