Real-time safety helmet detection using enhanced YOLOv5 object detection

Muhammad Zamri, Fatin Najihah and Zulkurnain, Nurul Fariza and Gunawan, Teddy Surya and Kartiwi, Mira and Md Yusoff, Nelidya and Nur, Levy Olivia (2024) Real-time safety helmet detection using enhanced YOLOv5 object detection. In: IEEE 9th International Conference on Smart Instrumentation, Measurement and Applications (ICSIMA 2023), 17th - 18th October 2023, Kuala Lumpur, Malaysia.

	PDF (Full Paper) - Published Version Restricted to Registered users only Download (1MB) | Request a copy
	PDF - Published Version Restricted to Registered users only Download (563kB) | Request a copy

Abstract

Personal Protective Equipment (PPE) regulations require construction workers to wear safety helmets to ensure site safety. However, monitoring PPE compliance consistently in fast-paced, dynamic construction environments poses a significant challenge. In response, we developed a sophisticated object detection system that automates the real-time verification of helmet use, thereby improving safety standards and reducing the likelihood of accidents. Extensive research was conducted to analyze all feasible algorithms that can be implemented in the safety helmet detection system and compare the proposed model with an existing one to ensure the proposed system can give high accuracy and high inference speed. Therefore, YOLOv5 was identified as the ideal choice in terms of accuracy and speed, and it was then enhanced with optimized transfer learning. We began our methodology by pre-training a comprehensive Kaggle dataset before refining the model using Roboflow on a specialized dataset. Using PyTorch and YOLOv5, we conducted exhaustive model training, testing, and evaluation. Our system achieved a lightning-fast inference speed of 39.8 milliseconds and a remarkable 91.4 percent accuracy in identifying helmet compliance. The implementation of such object detection technologies has the potential to significantly increase safety helmet compliance, thereby creating a safer environment for construction workers.

Item Type:	Proceeding Paper (Invited Papers)
Additional Information:	External collaboration: - National: UTM - International: Telkom University, Indonesia
Uncontrolled Keywords:	Personal Protective Equipment, YOLOv5 object detection, transfer learning, helmet compliance, real-time verification.
Subjects:	T Technology > TK Electrical engineering. Electronics Nuclear engineering > TK7800 Electronics. Computer engineering. Computer hardware. Photoelectronic devices > TK7885 Computer engineering
Kulliyyahs/Centres/Divisions/Institutes (Can select more than one option. Press CONTROL button):	Kulliyyah of Engineering Kulliyyah of Engineering > Department of Electrical and Computer Engineering Kulliyyah of Information and Communication Technology Kulliyyah of Information and Communication Technology Kulliyyah of Information and Communication Technology > Department of Information System Kulliyyah of Information and Communication Technology > Department of Information System
Depositing User:	Prof. Dr. Teddy Surya Gunawan
Date Deposited:	15 Jan 2024 10:29
Last Modified:	28 Feb 2024 13:13
URI:	http://irep.iium.edu.my/id/eprint/110164

Actions (login required)

View Item