Cigarette automatic detection method based on deep learning in monitoring scene

Abstract

The invention relates to a cigarette automatic detection method based on deep learning in a monitoring scene, and the method comprises the steps: firstly carrying out the overturning, zooming and smoothing of a cigarette data set photographed and downloaded from a network, obtaining a bigger data set, training the data set based on a YOLOv3 deep learning network, and forming a template library; performing corresponding image enhancement processing on the to-be-detected image or video frame by using an image enhancement method; carrying out image segmentation on the large-size image, cigarettesor people with cigarettes in the image are separated out, and the time needed by detection is shortened; then pre-generating prediction boxes on the image to be detected, and comparing each prediction box with a pre-trained template library; and finally, selecting a prediction box higher than a preset threshold value from the detection confidence coefficients of all the prediction boxes, and determining the prediction box as a target object. After the whole image of the current frame is scanned, all detected targets are marked and displayed on the image, and cigarette detection is completed.The method can effectively improve the detection accuracy and shorten the detection time.

Description

technical field [0001] The invention relates to the fields of deep learning and computer vision, in particular to an automatic cigarette detection method based on deep learning in a monitoring scene. Background technique [0002] With the continuous improvement of modern people's living standards, concepts continue to improve. People pay more and more attention to the dangers of smoking. Recently, accidents caused by smoking have been well-known, such as: smoking on the high-speed rail caused the suspension of the high-speed rail, causing hundreds of passengers to be stranded; smoking at the gas station caused the gas station to catch fire, resulting in casualties and property losses; more What's more, forest fires were caused by smoking, and the great rivers and mountains of the motherland were destroyed. Therefore, smoking is strictly prohibited on high-speed railways and EMUs. In recent years, the punishment has been increased to the legal level; in public areas such as...

AI Technical Summary

Problems solved by technology

Not only does it require additional expenses for the installation of related equipment, but it also cannot give certain warnings after the smoking event occurs

Images