Helmet wearing detection study based on P2DFE-YOLOv8n

Xin He; Wenzhong Zhu; Yu Liu; Jie Li

doi:10.1117/12.3033777

13 June 2024 Helmet wearing detection study based on P2DFE-YOLOv8n

Xin He, Wenzhong Zhu, Yu Liu, Jie Li

Proceedings Volume 13180, International Conference on Image, Signal Processing, and Pattern Recognition (ISPP 2024); 131804B (2024) https://doi.org/10.1117/12.3033777
Event: International Conference on Image, Signal Processing, and Pattern Recognition (ISPP 2024), 2024, Guangzhou, China

Abstract

To address the challenges of low detection accuracy and substantial model parameter size present in traditional safety helmet detection algorithms, this study introduces an enhanced algorithm known as P2DFE-YOLOv8n, specifically crafted for the detection of safety helmet usage. This algorithm builds upon the foundation of YOLOv8n by incorporating a specialized detection head for small targets, aimed at improving the accuracy of safety helmet detection. Simultaneously, it eliminates the detection head for large targets to reduce the model's parameter count and computational complexity, without compromising detection accuracy. The inclusion of the FasterNet Block and C2f modules in the model architecture serves to further streamline the model and enhance its accuracy. Additionally, the employment of the EMA (Efficient Multi-Scale Attention) attention mechanism within the neck networks allows for focused processing of crucial information related to the individual wearing the helmet, thereby significantly increasing detection precision. Experimental evaluations conducted on the proprietary Hhelmet safety gear database demonstrate that the refined P2DFE-YOLOv8n algorithm achieves an mAP@0.5 score of 92.3%, with a remarkably reduced parameter volume of just 0.76M. Compared to the original YOLOv8n model, the refined algorithm presents a reduction in parameter size and model size by 2.25M (74.8%) and 4.294MB (70.6%), respectively. Furthermore, the enhanced model showcases improvements in accuracy, recall rate, mAP@0.5, mAP@0.5:0.95, and FPS by 1.8%, 0.2%, 1.0%, 0.2%, and 14, respectively, underscoring the efficacy of the proposed modifications in optimizing both performance and efficiency in safety helmet detection tasks.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Xin He, Wenzhong Zhu, Yu Liu, and Jie Li "Helmet wearing detection study based on P2DFE-YOLOv8n", Proc. SPIE 13180, International Conference on Image, Signal Processing, and Pattern Recognition (ISPP 2024), 131804B (13 June 2024); https://doi.org/10.1117/12.3033777

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
8 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Target detection

Object detection

Head

Detection and tracking algorithms

Safety

Small targets

Data modeling

Show All Keywords

Keywords/Phrases

Search In:

Publication Years