An Improved YOLOv5 Real-time Detection Method for Aircraft Target Detection

Yuxi Chen,Jiaye Yang,Jianfeng Wang,Xue Zhou,Jianxiao Zou,Yun Li
DOI: https://doi.org/10.1109/icac55051.2022.9911114
2022-01-01
Abstract:To address the lack of accuracy and speed of aircraft detection in complex background images, the YOLOv5 model is improved in this paper to meet the growing demand for aircraft detection. Firstly, Mosaic-9 data enhancement is performed on the dataset, and MobileNet V3 Small is used to replace ResNet feature extraction network to enrich small target samples and improve feature extraction speed. Secondly, over-fitting is prevented by Label Smoothing, and channel pruning is used to reduce the complexity and redundancy of the neural network. Finally, TensorRT is used to optimize the underlying hardware resources to improve the detection speed. The experiments use Military Aircraft Detection Dataset for model training and validation on NVIDIA Tesla P100 16GB GPU. The results show that our improvements maintain the accuracy of the model while effectively increasing the training speed and detection speed of the model. Our codes are available at https://github.com/imcyx/Military-Detect
What problem does this paper attempt to address?