A lightweight face-assisted object detection model for welding helmet use

Automatic Welding Helmet Use (WHU) detection technology is of great significance for the safety management of construction site, then, this paper proposes a lightweight face-assisted model using YOLOv5s for the detection of WHU (WHU-YOLO). First, the Ghost module is introduced into YOLOv5s to optimi...

Full description

Saved in:
Bibliographic Details
Published in:Expert systems with applications 2023-07, Vol.221, p.119764, Article 119764
Main Authors: Chen, Weiming, Li, Changfan, Guo, Hailin
Format: Article
Language:English
Subjects:
Computer vision
Deep learning
Helmet detection
Object detection
Welding helmet
YOLOv5
Citations: Items that this one cites
Items that cite this one
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Automatic Welding Helmet Use (WHU) detection technology is of great significance for the safety management of construction site, then, this paper proposes a lightweight face-assisted model using YOLOv5s for the detection of WHU (WHU-YOLO). First, the Ghost module is introduced into YOLOv5s to optimize feature extraction parts of backbone and neck, reducing model complexity. Then, the neck of YOLOv5s is reconstructed based on Bi-directional Feature Pyramid Network (Bi-FPN). The experimental results implemented on established Welding helmet and Human face Detection (WHD) dataset indicate that the false positives have been greatly decreased with the assistance of face data and the mean average precision (mAP) reaches 83.65%. Meanwhile, under the environment of NVIDIA GeForce GTX 1070 and 640 × 640 input size, WHU-YOLO with inference time up to 5.7 ms achieves model compression with 35.7%, 34.4% and 30.1% reductions in parameters, weight size and Floating Point Operations (FLOPs) compared with YOLOv5s, respectively, which has no decline on detection performance.
ISSN:0957-4174
1873-6793
DOI:10.1016/j.eswa.2023.119764