Lightweight YOLO v4 security protection detection method based on attention mechanism improvement

The invention discloses a lightweight YOLO v4 security protection detection method based on attention mechanism improvement, and the method comprises the steps: collecting images, building a data set, employing a YOLO v4 network as a basic model, integrating an attention mechanism into a YOLO v4 mod...

Full description

Saved in:
Bibliographic Details
Main Authors WANG WEI, YUN JIAN, LIU YONGKUI, DUO JUNJIE, ZHANG JIANXIN, YU YANGYUE
Format Patent
LanguageChinese
English
Published 19.10.2021
Subjects
CALCULATING
COMPUTER SYSTEMS BASED ON SPECIFIC COMPUTATIONAL MODELS
COMPUTING
COUNTING
HANDLING RECORD CARRIERS
PHYSICS
PRESENTATION OF DATA
RECOGNITION OF DATA
RECORD CARRIERS
Online AccessGet full text

Cover

More Information
Summary:The invention discloses a lightweight YOLO v4 security protection detection method based on attention mechanism improvement, and the method comprises the steps: collecting images, building a data set, employing a YOLO v4 network as a basic model, integrating an attention mechanism into a YOLO v4 model, and obtaining an improved YOLO v4 security protection detection model; training the improved YOLO v4 safety protection detection model by using the data set to obtain an improved YOLO v4 target detection model; inputting a to-be-detected image into the improved YOLO v4 safety protection detection model, outputting corresponding target detection results by adopting the improved YOLO v4 target detection model, respectively positioning a pedestrian detection frame, a safety helmet detection frame and a safety reflective vest detection frame, and respectively calculating CIoU of three targets to obtain a final target detection frame; and detecting and identifying a construction scene safety equipment target, and ju
Bibliography:Application Number: CN202110782188