Deep learning-based detection method and device of safety-helmet wearing in video

Abstract

The invention discloses a deep learning-based detection method and device of safety-helmet wearing in a video. The method comprises: taking M pictures from collected video data; labeling N pictures inthe M pictures, and labeling the M pictures to obtain pedestrian positive-sample data, pedestrian negative-sample data, safety-helmet positive-sample data and safety-helmet negative-sample data; using the pedestrian positive-sample data and the pedestrian negative-sample data to train a pedestrian detection network, and using the safety-helmet positive-sample data and the safety-helmet negative-sample data to train a safety-helmet detection network; training a to-be-detected image, and obtaining a to-be-detected area through a background difference method; inputting the to-be-detected area tothe trained pedestrian detection network to judge whether a pedestrian exists in the to-be-detected area; if yes, executing an operation of inputting a preset area in the to-be-detected area, where the pedestrian exists, into the trained safety-helmet detection network for detection; and issuing an alarm if the pedestrian is not detected. By applying the embodiment of the invention, detection efficiency can be improved.

Description

technical field [0001] The present invention relates to a safety helmet detection method and device, and more specifically to a safety helmet wearing detection method and device in a video based on deep learning. Background technique [0002] Hard hats are required to be worn by miners, electricians, construction workers and other types of work, which can effectively protect the safety of personnel, but a series of accidents caused by not wearing hard hats have sounded the alarm for us. Not wearing a safety helmet itself will increase safety hazards. There are also a few people who like to sit on safety helmets as stools. A little carelessness may lead to safety hazards or accidents. In order to reduce the occurrence of such situations, surveillance cameras are installed in many working environments. Through video surveillance technology, it is possible to keep abreast of the wearing conditions of employees' safety helmets. To watch multiple sets of pictures at the same tim...

AI Technical Summary

Problems solved by technology

[0005] The technical problem to be solved by the present invention is to provide a method and device for detecting helmet wearing in videos based on deep learning, so as to solve the shortcomings of low detection efficiency in the prior art

Images