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...
Saved in:
Main Authors | , , , , , |
---|---|
Format | Patent |
Language | Chinese English |
Published |
19.10.2021
|
Subjects | |
Online Access | Get full text |
Cover
Loading…
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 |