关键词:
遥感图像
目标检测
小目标
无人机图像
多尺度目标
Transformer
多光谱图像
摘要:
遥感图像是通过传感器远距离捕获目标的电磁辐射经处理后形成的图像数据,不仅记录了地形、水体、植被等自然要素的空间分布与特征,还包含了人造目标及活动痕迹的关键信息,为多个学科领域的研究与应用提供了宝贵的数据支持。近年来,由于深度学习的迅猛发展,目标检测任务取得了显著的技术革新。基于深度学习的目标检测算法能够快速、准确地定位与识别图像中的感兴趣目标,为遥感图像中的关键信息提取提供了自动化和智能化的解决方案,在环境监测、农业生产、灾难营救、交通管控、智慧城市、国防安全等领域具有重要的研究意义和应用价值。
基于深度学习的遥感图像目标检测算法在多种场景和任务中展现了卓越的准确性和稳定性,但在特定复杂场景的应用中仍面临严峻挑战:遥感图像中的小目标特征信息稀少,易受背景和噪声干扰,分布密集,标注成本高;由于飞行高度、飞行速度以及拍摄角度的多样性,无人机图像中的目标呈现出尺度差异明显、空间分布不均匀和背景复杂多变的特点,同时存在遮挡和重叠问题以及运动模糊效应;在全天候遥感场景中,可见光图像在低光照环境中严重缺乏目标的视觉特征,导致检测精度受限,而红外图像因缺少细节信息,使得模型难以准确判断目标的类别。为了解决上述问题,本文以多场景遥感图像目标检测关键技术为研究主题,聚焦于探索遥感图像小目标检测、无人机图像多尺度目标检测以及全天候遥感场景多光谱图像目标检测,致力于推动遥感图像目标检测领域的理论创新与实际应用。本文的主要贡献如下:
1.针对遥感图像小目标检测的难点,提出了一种基于传统注意力机制的遥感图像小目标检测算法。通过整合通道注意力机制和卷积瓶颈模块,设计了新颖的颈部注意力模块NAB。该模块不仅能够动态增强特征和抑制噪声,而且摒弃了传统注意力作用于原始特征图的惯例,提升了注意力机制的灵活性和特征表示能力。在YOLOv3的基础上,通过引入NAB,并探索多种边界框损失函数对模型性能的影响,构建了遥感图像小目标检测模型YOLOv3-NAB。为了解决小目标检测数据集匮乏的问题,制作了一个高分辨率航空车辆小目标数据集VHRAI。在VHRAI数据集上,YOLOv3-NAB取得了89.3 m AP@50和36.7APs,较基准模型分别提升了3.4和5.6,同时有效降低了计算开销。在TGRS-HRRSD数据集上,YOLOv3-NAB展现了出色的检测性能和泛化能力。此外,针对实时检测任务,构建了轻量化目标检测模型YOLOv4-Tiny-NAB。在VHRAI和TGRS-HRRSD数据集上,该模型在保持高效计算特性的同时,显著提升了检测性能。
2.针对无人机场景中目标的特点,提出了一种基于自注意力机制的无人机图像多尺度目标检测算法。通过改进Transformer以适应图像的空间特性,设计了Convformer模块,并与卷积瓶颈模块相结合,构建了全局-局部信息融合模块GLIFM。该模块在建模远距离依赖关系的同时,保留了对局部特征的关注。在路径聚合网络的信息流动过程中,通过引入微小尺度特征图,为小尺度分支提供丰富的细节信息。在快速空间金字塔池化模块中,通过设计通道信息增强分支,进一步增强了模型的多尺度特征表示能力。基于上述创新方法,构建了无人机图像多尺度目标检测模型YOLO-UAV。通过在Vis Drone数据集上进行消融实验,验证了Convformer的有效性以及Transformer在处理图像任务的局限性。在与众多模型的对比中,YOLO-UAV仅用5.2M的参数量,取得了47.4 m AP@50和29.3m AP@50-95的优异表现。
3.针对全天候遥感场景中的挑战,提出了一种基于交叉注意力机制的多光谱图像目标检测算法。为了实现多光谱特征的充分融合与交互,设计了高效交叉注意力融合模块ECAFM。该模块采用了一种创新性的交叉注意力机制,通过建模不同光谱特征图对于融合特征图的远距离依赖关系,同时实现了同一光谱内的特征挖掘和不同光谱间的信息交互。在ECAFM中,还提出了局部信息增强的特征缩放方法,在显著降低计算复杂度的同时,缓解了降采样-上采样过程造成的信息损失,进一步增强了模型的特征表示能力。此外,为了扩大小尺度融合特征图的感受野以及提供丰富的语义信息,提出了小尺度信息增强模块SSIEM。该模块通过整合所有尺度的特征图并计算自注意力机制的方式,有效增强了小尺度融合特征图的特征表示能力。基于ECAFM和SSIEM,构建了轻量化多光谱图像目标检测模型YOLO-FT。在Drone Vehicle数据集上,YOLO-FT取得了83.3m AP@50和63.4 m AP@50-95,较YOLOv8s-Add分别提升了1.0和0.8,并且参数量减少了大约1/3。在LLVIP数据集上,YOLO-FT的各项指标超越了众多经典的多光谱图像目标检测模型,展现出卓越的检测性能和泛化