Affiliations: [a] Zhangjiakou Cigarette Factory Co., Ltd, Zhangjiakou, Hebei, China | [b] Information Engineering College, Hebei University of Architecture, Zhangjiakou, Hebei, China
Correspondence:
[*]
Corresponding author: Zeshan Han, Information Engineering College, Hebei University of Architecture, Zhangjiakou, Hebei, China. E-mail: 13931233247m0@sina.cn.
Abstract: The use of general target detection algorithms for small-target detection is computationally costly and has a high missed detection rate. A lightweight small-target detection model based on YOLOv5 is proposed to address this issue.First, a maximum pooling layer is introduced to reduce the number of calculations. Second, Shuffle_Conv is designed to replace the ordinary convolutional layer to reduce model parameters. To further compress the model, the Add fusion method is used in the C3 module, while the GAC3 layer is designed with GhostNet. Finally, Mosaic_9 is introduced to improve the small-target detection without increasing the number of calculations. Compared with YOLOv5, computation and parameters of the improved model are reduced by 84.9% and 39.1%, respectively, and the accuracy is improved by 2%, which is more obvious than that of the original model.
Keywords: Object detection, YOLOv5, lightweight, small targets
