Izvorni znanstveni članak
https://doi.org/10.20532/cit.2023.1005754
Research on Small Target Detection Method for Industrial Safety Helmets Based on Improved YOLOv8
Lan Yuanshuai
orcid.org/0009-0001-3289-8519
; School of Electronic Information Engineering, Geely University, Chengdu, Sichuan, China
Chen Mo
; School of Electronic Information Engineering, Geely University, Chengdu, Sichuan, China
Li Chuan
; Chengdu College, University of Electronic Science and Technology of China, Chengdu, Sichuan, China
Wang Qian
; School of Electronic Information Engineering, Geely University, Chengdu, Sichuan, China
Liao Min
; School of Electronic Information Engineering, Geely University, Chengdu, Sichuan, China
Sažetak
Industrial safety helmets are crucial personal protective gear but detecting them as small targets in complex environments is challenging. This work proposes enhancements to the YOLOv8 object detection framework, specifically incorporating a spatial-to-depth (SPD) convolution module and a large selective kernel network (LSKNet). SPD-Conv combines spatial-to- depth layers and non-strided convolutions to retain fine-grained information when downscaling feature maps, while LSKNet introduces dynamic spatial selection and attention for refined context modeling. Our customized model is trained on a dataset of construction hardhat images captured via drones. Quantitative results showcase higher precision and recall over baseline YOLOv8, surpassing competing YOLOv5 versions. An optimized final model outcomes demonstrate accuracy exceeding 90% validation in mAP metric after 200 training rounds. By tackling limitations posed by small, obscured industrial safety gears, this enhanced real-time detection approach provides indispensable technological support for bolstering workplace hazard identification and prevention.
Ključne riječi
Small Target Detection; Target Detection; safety helmet; YOLOv8
Hrčak ID:
314967
URI
Datum izdavanja:
27.2.2024.
Posjeta: 230 *