GCD-YOLOv5: An Armored Target Recognition Algorithm in Complex Environments Based on Array Lidar

Abstract

For the recognition of armored targets in complex battlefield environments, how to reduce missed and false alarms while achieving real-time is an urgent issue. To this end, the GCD-YOLOv5 algorithm is innovatively proposed. Firstly, array lidar is used to acquire the armor target data. Secondly, the armor target data is expanded with an improved GAN(Generative Adversarial Network) to increase the diversity of training data. Afterward, the expanded dataset is fed into the GCD-YOLv5(You Only Look Once) for training. And the GCD-YOLOv5 is reflected in the following aspects. Firstly, the CBAM(Convolutional Block Attention Module) and the multi-scale feature fusion are added to improve the feature extraction capability and detection efficiency, increasing the recognition capability of small and obscured targets. Secondly, combining with DETR(Detection Transformer) to lighten YOLOv5 to achieve the real-time requirement. Thirdly, the YOLOv5 loss function and prediction box filtering method are improved to increase the detection accuracy and the confidence of the detection boxes. The experimental results show that the GCD-YOLOv5 algorithm has higher accuracy and real-time, the mAP(mean Average Precision) can reach 99.7%, and fps is 68.56% higher compared to YOLOv5, which significantly improves the recognition capability of armored targets in complex battlefield environments.