Small Target Detection Algorithm in Aerial Images Based on Improved RT-DETR

Zi-qi ZHAO , Wei-dong LI , Xiao-juan LI

Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (13) : 5527 -5534.

PDF
Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (13) : 5527-5534 DOI: 10.12404/j.issn.1671-1815.2404266
research-article
Small Target Detection Algorithm in Aerial Images Based on Improved RT-DETR
Author information +
History +
PDF

Abstract

An algorithm has been proposed to detect small targets in unmanned aerial vehicle(UAV) aerial images. The algorithm is based on an improved real-time detection Transformer (RT-DETR) and aims to address the challenges posed by complex backgrounds and a large number of small target samples. To enhance the feature fusion network, a dedicated feature fusion structure for small targets has been incorporated, utilizing rich location information from the shallow feature map to improve the network's ability to detect small targets. Furthermore, the last residual block in the BackBone has been removed to prevent an increase in additional parameters. Additionally, the MCP Block, a reconstructed BasicBlock structure in the backbone network, has been designed, which includes a multi-channel feature partial convolution module (MCPConv) to reduce redundancy in channel features and enhance the acquisition of multi-scale detail features. Moreover, a location encoding mechanism with learning ability has been introduced to obtain more accurate and expressive location information. The normalized weighted deviation (NWD) and mean precision-driven IoU(MPDIoU) positioning loss functions have been incorporated to accelerate the convergence speed of the model and reduce sensitivity to position deviation. Experimental results on the VisDrone2019-DET dataset demonstrate that the improved model reduces parameters by 62% compared to the original model, increases mAP50 by 3.9%, and improves FPS by 17%. The improved model exhibits superior detection performance compared to other mainstream detection models.

Keywords

small object detection / RT-DETR / multi channel partial convolution / learned position embedding

Cite this article

Download citation ▾
Zi-qi ZHAO, Wei-dong LI, Xiao-juan LI. Small Target Detection Algorithm in Aerial Images Based on Improved RT-DETR. Science Technology and Engineering, 2025, 25(13): 5527-5534 DOI:10.12404/j.issn.1671-1815.2404266

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (7458KB)

17

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉