Affiliation:
1. Xi’an Research Institute of High-Tech, Xi’an 710025, China
Abstract
The detection of infrared vehicle targets by UAVs poses significant challenges in the presence of complex ground backgrounds, high target density, and a large proportion of small targets, which result in high false alarm rates. To alleviate these deficiencies, a novel YOLOv7-based, multi-scale target detection method for infrared vehicle targets is proposed, which is termed YOLO-ViT. Firstly, within the YOLOV7-based framework, the lightweight MobileViT network is incorporated as the feature extraction backbone network to fully extract the local and global features of the object and reduce the complexity of the model. Secondly, an innovative C3-PANet neural network structure is delicately designed, which adopts the CARAFE upsampling method to utilize the semantic information in the feature map and improve the model’s recognition accuracy of the target region. In conjunction with the C3 structure, the receptive field will be increased to enhance the network’s accuracy in recognizing small targets and model generalization ability. Finally, the K-means++ clustering method is utilized to optimize the anchor box size, leading to the design of anchor boxes better suited for detecting small infrared targets from UAVs, thereby improving detection efficiency. The present article showcases experimental findings attained through the use of the HIT-UAV public dataset. The results demonstrate that the enhanced YOLO-ViT approach, in comparison to the original method, achieves a reduction in the number of parameters by 49.9% and floating-point operations by 67.9%. Furthermore, the mean average precision (mAP) exhibits an improvement of 0.9% over the existing algorithm, reaching a value of 94.5%, which validates the effectiveness of the method for UAV infrared vehicle target detection.
Funder
National Natural Science Foundation of China
National Key Basic Research Strengthen Foundation of China
Subject
General Earth and Planetary Sciences
Reference53 articles.
1. Deep Learning for Unmanned Aerial Vehicle-Based Object Detection and Tracking: A survey;Wu;Geosci. Remote Sens.,2022
2. Qiu, Z., Bai, H., and Chen, T. (2023). Special Vehicle Detection from UAV Perspective via YOLO-GNS Based Deep Learning Network. Drones, 7.
3. YOLOv5-Based Vehicle Detection Method for High-Resolution UAV Images;Chen;Mob. Inf. Syst.,2022
4. SI-EDTL: Swarm intelligence ensemble deep transfer learning for multiple vehicle detection in UAV images;Shokouhifar;Concurr. Comput. Pract. Exp.,2021
5. Multi-UAV Search and Rescue with Enhanced A∗ Algorithm Path Planning in 3D Environment;Du;Int. J. Aerosp. Eng.,2023
Cited by
6 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献