YOLO-Cigarette: An effective YOLO Network for outdoor smoking Real-time Object Detection

Yankai Ma,Jun Yang,Zhendong Li,Ziqiang Ma
DOI: https://doi.org/10.1109/CBD54617.2021.00029
2022-01-01
Abstract:Smoking behavior is the most important factor that causes fire, most of the traditional smoking detection is to detect the smoke produced by the burning of cigarettes,but it is not applicable in public places with strong air circulation. This paper designs a smoking image detection model:YOLO-Cigarette,which is based on deep learning, it solves the problems of low detection accuracy caused by small cigarette targets and complex background environment in the existing cigarette detection algorithms. The original YOLOv5 network structure is improved by fusing the new fine-grained spatial pyramid pooling module(FSPP) and multi spatial attention mechanism (MSAM), effectively improving the accuracy of small target detection. At the same time, the parameters in the model are quantized to greatly reduce the calculation of the model and improve the detection speed. YOLO-Cigarette model is 4.8% better than the original model in map, and achieves 95% accuracy in the test set.
What problem does this paper attempt to address?