Ke Xiao, Chen Wenyao, Guo Wenzhong
Fujian Provincial Key Laboratory of Networking Computing and Intelligent Information Processing, College of Mathematics and Computer Science, Fuzhou University, Fuzhou, 350116 China.
Key Laboratory of Spatial Data Mining & Information Sharing, Ministry of Education, Fuzhou, 350003 China.
Peer Peer Netw Appl. 2022;15(2):950-972. doi: 10.1007/s12083-021-01258-4. Epub 2021 Nov 15.
In industrial production, personal protective equipment (PPE) protects workers from accidental injuries. However, wearing PPE is not strictly enforced among workers due to all kinds of reasons. To enhance the monitoring of workers and thus avoid safety accidents, it is essential to design an automatic detection method for PPE. In this paper, we constructed a dataset called FZU-PPE for our study, which contains four types of PPE (helmet, safety vest, mask, and gloves). To reduce the model size and resource consumption, we propose a lightweight object detection method based on deep learning for superfast detection of whether workers are wearing PPE or not. We use two lightweight methods to optimize the network structure of the object detection algorithm to reduce the computational effort and parameters of the detection model by 32% and 25%, respectively, with minimal accuracy loss. We propose a channel pruning algorithm based on the BN layer scaling factor to further reduce the size of the detection model. Experiments show that the automatic detection of PPE using our lightweight object detection method takes only 9.5 ms to detect a single video frame and achieves a detection speed of 105 FPS. Our detection model has a minimum size of 1.82 MB and a model size compression rate of 86.7%, which can meet the strict requirements of memory occupation and computational resources for embedded and mobile devices. Our approach is a superfast detection method for green edge computing.
在工业生产中,个人防护装备(PPE)可保护工人免受意外伤害。然而,由于各种原因,工人佩戴PPE的情况并未得到严格执行。为了加强对工人的监测,从而避免安全事故,设计一种PPE自动检测方法至关重要。在本文中,我们构建了一个名为FZU - PPE的数据集用于研究,该数据集包含四种类型的PPE(头盔、安全背心、口罩和手套)。为了减小模型大小和资源消耗,我们提出一种基于深度学习的轻量级目标检测方法,用于超快速检测工人是否佩戴PPE。我们使用两种轻量级方法优化目标检测算法的网络结构,分别将检测模型的计算量和参数减少32%和25%,同时精度损失最小。我们提出一种基于BN层缩放因子的通道剪枝算法,以进一步减小检测模型的大小。实验表明,使用我们的轻量级目标检测方法自动检测PPE,检测单个视频帧仅需9.5毫秒,检测速度达到105帧每秒。我们的检测模型最小尺寸为1.82MB,模型大小压缩率为86.7%,能够满足嵌入式和移动设备对内存占用和计算资源的严格要求。我们的方法是一种用于绿色边缘计算的超快速检测方法。