Improving the Accuracy of UAV Detection Through Combination of Different Convolution Units
The YOLO (You Only Look Once) series is widely used in various detection fields. This article applies the latest released YOLOv9 and YOLOv10 models to surface-to-air/airto-air UAV (Unmanned Aerial Vehicle) target detection tasks, assisting anti-UAV tasks and other airborne threat detection tasks. We propose selecting different convolution modules in network layers corresponding to various scales. Specifically, this article uses the DroneDetection dataset, expands the training set using data augmentation techniques, and trains multiple network models using transfer learning techniques. The experimental results show that compared with the most widely used YOLOv3 and YOLOv5 models, YOLOv9 and YOLOv5 models have significant advantages in speed due to their simplification of the network backbone, but have a significant decrease in detection accuracy represented by mAP 50-95 values. In addition, the combination of different convolution modules can improve the accuracy of UAV small target detection.
Authors
- Yucan Chi
- Chengchao Bai (ORCID: https://orcid.org/0000-0002-0349-9869)
- Jifeng Guo (ORCID: https://orcid.org/0000-0003-2840-9820)
Institutions
- Harbin Institute of Technology (CN)
Publication Details
- Published
- 2024-10-18
- DOI
- https://doi.org/10.1109/icus61736.2024.10839781
- Primary Topic
- Infrared Target Detection Methodologies
- Type
- article
- Field-Weighted Citation Impact
- 0.00