Vehicle Small Target Detection Algorithm for UAV Remote Sensing Images Based on YOLOv5

Jun-qing BAI , Meng-ting WANG , Shou-ting SHEN

Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (12) : 5110 -5118.

PDF
Science Technology and Engineering ›› 2025, Vol. 25 ›› Issue (12) : 5110-5118 DOI: 10.12404/j.issn.1671-1815.2403707
research-article
Vehicle Small Target Detection Algorithm for UAV Remote Sensing Images Based on YOLOv5
Author information +
History +
PDF

Abstract

Remote sensing images are characterized by diverse scales, dense arrangement and small target sizes, etc. Aiming at the problem that there is much background noise in remote sensing images and vehicle targets are small and difficult to be acquired. A vehicle target detection algorithm based on improved feature fusion method, Atiny-YOLO was proposed. Firstly, an additional detection layer for small targets was introduced into the Neck layer of YOLOv5 so as to generated a small target detection algorithm for drone remote sensing images. Neck layer to introduce an additional detection layer for small targets, so as to generated a larger-scale feature map and effectively identified the detailed features of small objects. Secondly, a split operation was added to the C3 module to reuse the image feature information, and the Swin Transformer module was further optimized to improve the usage rate of the effective information. Lastly, by improving the feature fusion channel, the detection accuracy was improved while the model parameters were reducing the model parameters. The Atiny-YOLO algorithm was tested on the AU-AIR (aerial universal autonomous inspection and recognition) dataset. The experimental results show that the average detection accuracy of the Atiny-YOLO algorithm compared to the baseline algorithm is improved by about 2.9%. It reaches 95.5% and the detection speed reaches 234 frames/s. These results verify that the Atiny-YOLO algorithm meets the real-time performance while the model detection accuracy is greatly improved.

Keywords

remote sensing images / Swin Transformer / vehicle detection / AU-AIR / Atiny-YOLO

Cite this article

Download citation ▾
Jun-qing BAI, Meng-ting WANG, Shou-ting SHEN. Vehicle Small Target Detection Algorithm for UAV Remote Sensing Images Based on YOLOv5. Science Technology and Engineering, 2025, 25(12): 5110-5118 DOI:10.12404/j.issn.1671-1815.2403707

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (14550KB)

18

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉