Command Palette
Search for a command to run...
Large, Complex, and Realistic Safety Clothing and Helmet Detection: Dataset and Method
Fusheng Yu Jiang Li Xiaoping Wang Shaojin Wu Junjie Zhang Zhigang Zeng

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 https://github.com/lijfrank-open/SFCHD-SCALE.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| object-detection-on-sfchd | SSD | mAP@0.50: 72.8 mAP@0.5:0.95: 41.5 |
| object-detection-on-sfchd | VFNet | mAP@0.50: 76.4 mAP@0.5:0.95: 51.0 |
| object-detection-on-sfchd | Faster RCNN | mAP@0.50: 76.4 mAP@0.5:0.95: 50.3 |
| object-detection-on-sfchd | TOOD+SCALE | mAP@0.50: 79.3 mAP@0.5:0.95: 52.4 |
| object-detection-on-sfchd | FCOS+SCALE | mAP@0.50: 76.3 mAP@0.5:0.95: 49.5 |
| object-detection-on-sfchd | YOLOv8+SCALE | mAP@0.50: 78.6 mAP@0.5:0.95: 53.3 |
| object-detection-on-sfchd | FCOS | mAP@0.50: 76.4 mAP@0.5:0.95: 49.6 |
| object-detection-on-sfchd | TOOD | mAP@0.50: 78.9 mAP@0.5:0.95: 52.3 |
| object-detection-on-sfchd | RetinaNet | mAP@0.50: 75.9 mAP@0.5:0.95: 48.9 |
| object-detection-on-sfchd | VFNet+SCALE | mAP@0.50: 76.6 mAP@0.5:0.95: 51.4 |
| object-detection-on-sfchd | YOLOv8 | mAP@0.50: 77.9 mAP@0.5:0.95: 52.2 |
| object-detection-on-sfchd | YOLOv5 | mAP@0.50: 74.1 mAP@0.5:0.95: 49.6 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.