Large, Complex, and Realistic Safety Clothing and Helmet Detection: Dataset and Method

Fusheng Yu,Jiang Li,Xiaoping Wang,Shaojin Wu,Junjie Zhang,Zhigang Zeng

2024-08-11

Abstract:Detecting safety clothing and helmets is paramount for ensuring the safety of construction workers. However, the development of deep learning models in this domain has been impeded by the scarcity of high-quality datasets. In this study, we construct a large, complex, and realistic safety clothing and helmet detection (SFCHD) dataset. SFCHD is derived from two authentic chemical plants, comprising 12,373 images, 7 categories, and 50,552 annotations. We partition the SFCHD dataset into training and testing sets with a ratio of 4:1 and validate its utility by applying several classic object detection algorithms. Furthermore, drawing inspiration from spatial and channel attention mechanisms, we design a spatial and channel attention-based low-light enhancement (SCALE) module. SCALE is a plug-and-play component with a high degree of flexibility. Extensive evaluations of the SCALE module on both the ExDark and SFCHD datasets have empirically demonstrated its efficacy in enhancing the performance of detectors under low-light conditions. The dataset and code are publicly available at <a class="link-external link-https" href="https://github.com/lijfrank-open/SFCHD-SCALE" rel="external noopener nofollow">this https URL</a>.

Computer Vision and Pattern Recognition

What problem does this paper attempt to address?

### Problems Addressed by the Paper

The paper primarily addresses the following issues:

1. **Insufficient Dataset**:
- Currently, the task of detecting safety clothing and helmets on construction sites and factories is severely hindered by the lack of high-quality datasets. Existing datasets (such as Pictor-v3 and SHWD) mainly focus on helmet detection while neglecting the importance of safety clothing, which may lead to safety hazards in practical applications.

2. **Overly Simplified Background**:
- Existing datasets have overly simplified backgrounds and idealized lighting conditions, which differ significantly from actual working environments. This results in models trained on these datasets having insufficient generalization ability in real-world applications.

3. **Poor Detection Performance Under Low Light Conditions**:
- In low light environments, the visibility of images is poor, severely affecting the performance of detectors. To address this issue, the authors designed a low-light enhancement module (SCALE) that combines spatial and channel attention mechanisms to improve the model's detection performance under low light conditions.

### Summary

This paper aims to enhance safety monitoring in actual industrial scenarios by constructing a large, complex, and realistic safety clothing and helmet detection dataset (SFCHD) and proposing a new low-light enhancement module (SCALE). These methods not only address the limitations of existing datasets but also significantly improve the performance of detection models under extreme lighting conditions.