基于YOLOv5的无人机遥感图像车辆小目标检测算法

白俊卿 ,  王梦婷 ,  沈守婷

科学技术与工程 ›› 2025, Vol. 25 ›› Issue (12) :5110-5118

PDF (14550KB)
科学技术与工程 ›› 2025, Vol. 25 ›› Issue (12) :5110-5118 DOI: 10.12404/j.issn.1671-1815.2403707
自动化技术、计算机技术

基于YOLOv5的无人机遥感图像车辆小目标检测算法

作者信息 +

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

Author information +
文章历史 +

摘要

遥感图像具有尺度多样、密集排列和目标尺寸小等特点,针对遥感图像中背景噪声多,车辆目标较小难以获取的问题,提出一种基于改善特征融合方法的车辆目标检测算法Atiny-YOLO。首先,在YOLOv5的Neck层中引入针对小目标的额外检测层,从而生成更大规模的特征图,有效识别小物体的细节特征;其次,向C3模块中添加split操作以复用图像特征信息,进一步优化Swin Transformer模块提高有效信息的使用率;最后,通过改善特征融合通道,提升检测精度的同时减少模型参数。在无人机视角数据集(aerial universal autonomous inspection and recognition, AU-AIR)数据集上验证Atiny-YOLO模型的有效性。实验结果表明:Atiny-YOLO算法相较于基线算法的平均检测精度提高了约2.9%。达到95.5%,检测速度达到234帧/s。这些结果验证了Atiny-YOLO算法在满足实时性的同时,模型检测精度大幅提升。

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.

关键词

遥感图像 / Swin Transformer / 车辆检测 / AU-AIR / Atiny-YOLO

Key words

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

引用本文

引用格式 ▾
白俊卿, 王梦婷, 沈守婷. 基于YOLOv5的无人机遥感图像车辆小目标检测算法[J]. 科学技术与工程, 2025, 25(12): 5110-5118 DOI:10.12404/j.issn.1671-1815.2403707
BAI Jun-qing, WANG Meng-ting, SHEN Shou-ting. Vehicle Small Target Detection Algorithm for UAV Remote Sensing Images Based on YOLOv5[J]. Science Technology and Engineering, 2025, 25(12): 5110-5118 DOI:10.12404/j.issn.1671-1815.2403707

登录浏览全文

4963

注册一个新账户 忘记密码

参考文献

基金资助

陕西省自然科学基金基础研究计划项目(2023-JC-YB-601)

西安市科技计划高校院所人才服务企业项目(23GXFW0077)

西安石油大学研究生精品课程建设项目(2023-X-YKC-003)

PDF (14550KB)

35

访问

0

被引

详细

导航
相关文章

/

〈 〉