Academic Journal of Computing & Information Science, 2023, 6(2); doi: 10.25236/AJCIS.2023.060209.
Yingying Cao, Mingkun Xu
School of Computer Science (National Pilot Software Engineering School), Beijing University of Posts and Telecommunications, Beijing, China
Aiming at the current problems in smoking detection in public places, such as many small and medium-sized cigarette targets, low pixels occupied by cigarette targets, indistinct characteristics of cigarette targets, and difficult detection, an improved smoking detection algorithm YOLOv5_EC is proposed. Based on the YOLOv5, this paper redesigns the neck structure and proposes the EFFN (Enhanced Feature Fusion Neck) structure, which fuses three adjacent feature maps of different sizes to better retain the positioning information of the target, further improves the feature expression ability of small objects, and then introduce the CBAM attention module in the network, so that the model can focus on the important areas of the image, suppress the interference of irrelevant information, enhances the model's ability to learn features, and improves the accuracy of model detection. Experiments show that the [email protected] of the model proposed in this paper is improved by 1.5% compared with the original YOLOv5 model, and the improved model can effectively identify smoking behavior in actual scenes.
smoking detection, YOLOv5 model, C3_CBAM, EFFN
Yingying Cao, Mingkun Xu. Smoking Detection Algorithm Based on Improved YOLOv5. Academic Journal of Computing & Information Science (2023), Vol. 6, Issue 2: 64-72. https://doi.org/10.25236/AJCIS.2023.060209.
[1] H. Tian, W. Li, L. Wang and P. Ogunbona, A Novel Video-Based Smoke Detection Method Using Image Separation[C]. 2012 IEEE International Conference on Multimedia and Expo, Melbourne, VIC, Australia, 2012, 532-537.
[2] Lisu Han, Leilei Rong, Yongquan Li, Zhikang Qin, and Yan Xu. CA-SSD-Based Real-time Smoking Target Detection Algorithm[C]. 2021 5th International Conference on Digital Signal Processing. Association for Computing Machinery, New York, NY, USA, 2021, 283–288.
[3] Shuai L, Bo W, Ranran D, et al. A novel smoke detection algorithm based on Fast Self-tuning background subtraction[C] IEEE 2016 Chinese Control and Decision Conference (CCDC) - Yinchuan, China, 2016, 3539-3543.
[4] REDMON J, DIVVALA S, GIRSHICK R, et al. You Only Look Once: Unified, Real-Time Object Detection[C] 2016 IEEE Conference on Computer Vision and Pattern Recognition(CVPR). LasVegas: IEEE, 2016, 779-788.
[5] LIU W, ANGUELOV D, ERHAN D, et al. SSD:Single Shot MultiBox Detector[C].Computer Vision–ECCV 2016. Amsterdam:Springer, Cham, 2016,21-37.
[6] GIRSHICK R, DONAHUE J, DARRELL T, et al.Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation[C].2014 IEEE Conference on Computer Vision and Pattern Recognition. Columbus:IEEE,2014:580-587.
[7] RENS Q, HEK M, GIRSHICK R,et al. Faster R-CNN:Towards Real-Time Object Detection with Region Proposal Networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, 39(6): 1137-1149.
[8] Poonam G, Shashank B N, Rao A G. Development of framework for detecting smoking scene in video clips[J]. Indonesian Journal of Electrical Engineering and Computer Science, 2019, 13(1):22-26.
[9] Z. Rentao, W. Mengyi, Z. Zilong, L. Ping and Z. Qingyu, Indoor Smoking Behavior Detection Based on YOLOv3-tiny[C]. 2019 Chinese Automation Congress (CAC), Hangzhou, China, 2019, 3477-3481.