基于改进RT-DETR的航拍图像小目标检测算法OA
Small object detection algorithm for aerial images based on improved RT-DETR
为了解决航拍图像中小目标检测精度低、模型参数量大及在复杂背景下特征提取能力不足等问题,提出了一种针对航拍图像小目标的轻量快速实时检测变换器(quick-enhanced real-time detection transformer,QRT-DETR)模型.该模型以实时检测变换器(real-time detection trans-former,RT-DETR)为基础,在主干网络中设计了轻量级ContextFocus Net主干网络,其核心的ELGC-Mixer模块采用双路径结构结合门控融合机制,分别提取并融合局部细节与全局上下文信息;在特征融合网络中,提出了高频增强跨尺度融合模块,通过利用高分辨率特征层P2与无参数SimAM注意力机制,构建了一条专属的高频信息通路;引入了一种融合形状感知与分布建模的联合损失函数,结合Inner-ShapeIoU损失函数与归一化 Wasserstein距离共同优化边界框回归.在VisDrone2019数据集上进行消融实验,并与其他模型进行性能对比.结果表明:QRT-DETR模型的mAP@50和mAP@50-95分别达到52.0%和32.6%,相较于原始RT-DETR模型分别提高了5.1个百分点和4.0个百分点,参数量减少了41.8%.所提模型可有效提升复杂场景下小目标的检测精度与模型轻量化水平,可为无人机航拍图像小目标检测任务提供可行的技术方案.
To address the problems of low detection accuracy,high model parameters,and insufficient feature extraction in complex backgrounds for small object detection in aerial images,a quick-enhanced real-time detection transformer(QRT-DETR)model for small object in aerial images was proposed.Built upon real-time detection transformer(RT-DETR),the model introduced a lightweight ContextFocus Net in the backbone,whose core ELGC-Mixer module employed a dual-path structure with a gating fusion mechanism to extract and fuse local details and global context information separately.In the feature fusion network,a high-frequency enhanced cross-scale fusion module was designed,utilizing a high-resolution P2 layer and parameter-free SimAM attention to construct a dedicated high-frequency information pathway.A joint loss function combining shape-aware Inner-ShapeIoU with normalized Wasserstein distance was introduced to optimize bounding box regression.Ablation experiments and performance comparisons with other models were conducted on the VisDrone2019 dataset.The results show that QRT-DETR achieves 52.0%mAP@50 and 32.6%mAP@50-95,improving by 5.1 percentage pointsand 4.0 percentage points over the original RT-DETR,respectively,while reducing parameters by 41.8%.The proposed model effectively enhances the detection accuracy for small objects in complex scenes and achieves lightweight design,providing a feasible technical solution for real-time small object detection in UAV aerial imagery.
王建霞;齐晨;张晓明
河北科技大学信息科学与工程学院,河北石家庄 050018河北科技大学信息科学与工程学院,河北石家庄 050018河北科技大学信息科学与工程学院,河北石家庄 050018
信息技术与安全科学
模式识别RT-DETR航拍图像小目标检测特征融合损失函数
pattern recognitionRT-DETRaerial imagessmall object detectionfeature fusionloss function
《河北工业科技》 2026 (3)
205-214,10
河北省自然科学基金(F2022208002)
评论