Safety helmet identification method and device based on YOLOv3, medium and equipment

A recognition method and helmet technology, which is applied in the field of image recognition, can solve the problems that the recognition algorithm cannot meet the actual needs of helmet recognition, and achieve the effects of meeting real-time detection requirements, improving safety assurance, improving accuracy and detection speed

Pending Publication Date: 2021-06-29
PINGAN PUHUI ENTERPRISE MANAGEMENT CO LTD
View PDF4 Cites 3 Cited by
  • Summary
  • Abstract
  • Description
  • Claims
  • Application Information

AI Technical Summary

Problems solved by technology

[0005] In order to solve the technical problem that the recognition algorithm in the prior art cannot meet the actual needs of hard hat recognition in safe production, the present disclosure provides a hard hat recognition method based on YOLOv3, including:

Method used

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
View more

Image

Smart Image Click on the blue labels to locate them in the text.
Viewing Examples
Smart Image
  • Safety helmet identification method and device based on YOLOv3, medium and equipment
  • Safety helmet identification method and device based on YOLOv3, medium and equipment
  • Safety helmet identification method and device based on YOLOv3, medium and equipment

Examples

Experimental program
Comparison scheme
Effect test

Embodiment 1

[0043] Such as figure 1 Shown:

[0044] The disclosure provides a YOLOv3-based helmet identification method, including:

[0045] S101: collecting construction site image data;

[0046] S102: Use an image annotation tool to annotate the image data to obtain annotated image data;

[0047] S103: Divide the labeled image data set into two parts, a training set and a test set, and divide the image data of the training set into different groups according to different resolutions;

[0048] S104: Utilize the neural network based on the Darknet-53 network model for model training, and obtain a trained helmet recognition model;

[0049] S105: Test the recognition accuracy of the helmet recognition model by using the test set image data through the YOLOv3 algorithm.

[0050] Further, the image annotation tool is a deep learning image annotation labelme tool.

[0051] LabelMe is a Javascript labeling tool for online image labeling. The advantage over traditional image annotation too...

Embodiment 2

[0094] Such as figure 2 as shown,

[0095] The present disclosure can also provide a helmet recognition device based on YOLOv3, including:

[0096] An image acquisition module 201, configured to acquire target image data;

[0097] An image labeling module 202, configured to use an image labeling tool to label the image data to obtain labeled image data;

[0098] An image grouping module 203, configured to divide the marked image data set into two parts, a training set and a test set, and divide the image data of the training set into different groups according to different resolutions;

[0099]The model training module 204 is used to select no less than one group of image data from the different groups of image data, and use the neural network based on Darknet-53 to perform model training on the image data of the training set , get the helmet recognition model after training;

[0100] A model testing module 205, configured to use the test set image data to test the recogn...

Embodiment 3

[0117] The present disclosure can also provide a computer storage medium on which a computer program is stored, and when the computer program is executed by a processor, it is used to realize the steps of the above-mentioned YOLOv3-based helmet identification method.

[0118] The computer storage medium of the present disclosure may be realized by using a semiconductor memory or a magnetic core memory.

[0119] Semiconductor memory, mainly used in computers, mainly has two types of semiconductor memory elements: Mos and bipolar. Mos components are highly integrated, the process is simple but the speed is slow. Bipolar components are complex in process, high in power consumption, low in integration but fast in speed. After the advent of NMos and CMos, Mos memory began to play a major role in semiconductor memory. NMos is fast, for example, the access time of Intel's 1K-bit SRAM is 45ns. CMos consumes less power, and the 4K-bit CMos static memory access time is 300ns. The ab...

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to view more

PUM

No PUM Login to view more

Abstract

The invention relates to the technical field of image recognition, and provides a safety helmet recognition method and device based on YOLOv3, a medium and equipment, and the method comprises the steps: collecting target image data; annotating the image data by using an image annotation tool to obtain annotated image data; dividing the marked image data set into a training set and a test set, and dividing the image data of the training set into different groups according to different resolutions; selecting at least one group of image data from the different groups of image data, and performing model training on the image data of the training set by using a neural network taking Darknet-53 as a basic network model to obtain a trained safety helmet recognition model; and testing an identification result of the safety helmet identification model by using the test set image data through a YOLOv3 algorithm. According to the method, data set classification and multi-scale and multi-level training are carried out, the accuracy and the detection speed of safety helmet detection are improved, and the real-time detection requirement is met.

Description

technical field [0001] The present disclosure relates to the technical field of image recognition, and more specifically, the present disclosure relates to a YOLOv3-based helmet recognition method, device, medium and equipment. Background technique [0002] Hard hats are an important protective measure to protect workers from accidental injuries in production and construction sites. Due to the lag of security measures, most of them rely on manual supervision. With the development of deep learning technology, good results have been achieved in various fields such as image recognition, speech recognition, and natural language processing. Image recognition is a very important research direction in the field of computer vision. In a complex construction environment, the accuracy of whether to wear a helmet needs to be improved. [0003] In recent years, people have paid more and more attention to the development of target detection and production safety. Some researchers have m...

Claims

the structure of the environmentally friendly knitted fabric provided by the present invention; figure 2 Flow chart of the yarn wrapping machine for environmentally friendly knitted fabrics and storage devices; image 3 Is the parameter map of the yarn covering machine
Login to view more

Application Information

Patent Timeline
no application Login to view more
Patent Type & Authority Applications(China)
IPC IPC(8): G06K9/00G06K9/62G06N3/04G06N3/08
CPCG06N3/08G06V20/52G06V2201/07G06N3/045G06F18/23G06F18/214
Inventor 李妍妍
Owner PINGAN PUHUI ENTERPRISE MANAGEMENT CO LTD
Who we serve
  • R&D Engineer
  • R&D Manager
  • IP Professional
Why Eureka
  • Industry Leading Data Capabilities
  • Powerful AI technology
  • Patent DNA Extraction
Social media
Try Eureka
PatSnap group products