Affiliation:
1. Zhangjiakou Cigarette Factory Co., Ltd, Zhangjiakou, Hebei, China
2. Information Engineering College, Hebei University of Architecture, Zhangjiakou, Hebei, China
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.
Reference24 articles.
1. Working With Fire Departments to Adapt and Implement Evidence-Based Programs That Increase Uptake of Smoke Alarms: A Case-Series Report;Omaki;J Burn Care Res.,2022
2. A fire alarm judgment method using multiple smoke alarms based on Bayesian estimation;Liu;Fire Saf J.,2023
3. Deep learning based object detection for resource constrained devices: Systematic review, future trends and challenges ahead;Kamath;Neurocomputing.,2023
4. Howard AG, Zhu M, Chen B, Kalenichenko D, Wang W, Weyand T, et al. MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications [Internet]. arXiv 2017 [cited 2024 Jan 23]. Available from: http//arxiv.org/abs/1704.04861.
5. Zhang X, Zhou X, Lin M, Sun J. ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices. In 2018 [cited 2024 Jan 23]. p. 6848-56. Available from: https//openaccess.thecvf.com/content_cvpr_2018/html/Zhang_ShuffleNet_An_Extremely_CVPR_2018_paper.html.