A smoking behavior detection method based on the YOLOv5 network

Smoking in public places not only brings about some safety hazards, but also does harm to people’s lives, property and living environment. A smoking behavior detection model based on deep learning is trained for the concern of environment and safety. First, a vertical rotation data enhancement method is adopted in the preprocessing stage to extend the dataset and increase the objects of detection. Then, the channel attention module is introduced in backbone network to calibrate the feature response. Finally, added a small target detection layer to the YOLOv5 algorithm. This paper analyzes the network structure of the YOLOv5s, and the model is trained and tested by utilizing the YOLOv5s network. Experimental results show that the mAP value of the algorithm is improved by 5.3% over the original algorithm.

[1]  Jiaming Qi,et al.  A Real-Time Detection Algorithm for Kiwifruit Defects Based on YOLOv5 , 2021, Electronics.

[2]  Darong Huang,et al.  Electrical Insulator Defects Detection Method Based on YOLOv5 , 2021, 2021 IEEE 10th Data Driven Control and Learning Systems Conference (DDCLS).

[3]  Xiaohan Shi,et al.  Detection of Flying Birds in Airport Monitoring Based on Improved YOLOv5 , 2021, 2021 6th International Conference on Intelligent Computing and Signal Processing (ICSP).

[4]  Cuiyu Wang,et al.  A Real-Time Smoke Detection Model Based on YOLO-SMOKE Algorithm , 2020, 2020 Cross Strait Radio Science & Wireless Technology Conference (CSRSWTC).

[5]  Jianhao Liao,et al.  Smoking target detection based on Yolo V3 , 2020, 2020 5th International Conference on Mechanical, Control and Computer Engineering (ICMCCE).

[6]  J. McClure,et al.  The United States National Cancer Institute’s Coordinated Research Effort on Tobacco Use as a Major Cause of Morbidity and Mortality among People with HIV , 2020, Nicotine & tobacco research : official journal of the Society for Research on Nicotine and Tobacco.

[7]  Hong-Yuan Mark Liao,et al.  YOLOv4: Optimal Speed and Accuracy of Object Detection , 2020, ArXiv.

[8]  Wang Mengyi,et al.  Indoor Smoking Behavior Detection Based on YOLOv3-tiny , 2019, 2019 Chinese Automation Congress (CAC).

[9]  Edward Sazonov,et al.  Smoking detection based on regularity analysis of hand to mouth gestures , 2019, Biomed. Signal Process. Control..

[10]  Chen Li,et al.  A SSD-based Crowded Pedestrian Detection Method , 2018, 2018 International Conference on Control, Automation and Information Sciences (ICCAIS).

[11]  Gang Sun,et al.  Squeeze-and-Excitation Networks , 2017, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[12]  Wei Liu,et al.  SSD: Single Shot MultiBox Detector , 2015, ECCV.

[13]  Ali Farhadi,et al.  You Only Look Once: Unified, Real-Time Object Detection , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[14]  Jian Sun,et al.  Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition , 2014, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[15]  Shau-Yin Tseng,et al.  Human Smoking Event Detection Using Visual Interaction Clues , 2010, 2010 20th International Conference on Pattern Recognition.

[16]  Young-Guk Ha,et al.  Image Preprocessing for Efficient Training of YOLO Deep Learning Networks , 2018, 2018 IEEE International Conference on Big Data and Smart Computing (BigComp).