河北工业科技2026,Vol.43Issue(3):205-214,10.DOI:10.7535/hbgykj.2026yx03002
基于改进RT-DETR的航拍图像小目标检测算法
Small object detection algorithm for aerial images based on improved RT-DETR
摘要
Abstract
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.关键词
模式识别/RT-DETR/航拍图像/小目标检测/特征融合/损失函数Key words
pattern recognition/RT-DETR/aerial images/small object detection/feature fusion/loss function分类
信息技术与安全科学引用本文复制引用
王建霞,齐晨,张晓明..基于改进RT-DETR的航拍图像小目标检测算法[J].河北工业科技,2026,43(3):205-214,10.基金项目
河北省自然科学基金(F2022208002) (F2022208002)