HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

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

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

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

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

lijfrank-open/SFCHD-SCALE
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
object-detection-on-sfchdSSD
mAP@0.50: 72.8
mAP@0.5:0.95: 41.5
object-detection-on-sfchdVFNet
mAP@0.50: 76.4
mAP@0.5:0.95: 51.0
object-detection-on-sfchdFaster RCNN
mAP@0.50: 76.4
mAP@0.5:0.95: 50.3
object-detection-on-sfchdTOOD+SCALE
mAP@0.50: 79.3
mAP@0.5:0.95: 52.4
object-detection-on-sfchdFCOS+SCALE
mAP@0.50: 76.3
mAP@0.5:0.95: 49.5
object-detection-on-sfchdYOLOv8+SCALE
mAP@0.50: 78.6
mAP@0.5:0.95: 53.3
object-detection-on-sfchdFCOS
mAP@0.50: 76.4
mAP@0.5:0.95: 49.6
object-detection-on-sfchdTOOD
mAP@0.50: 78.9
mAP@0.5:0.95: 52.3
object-detection-on-sfchdRetinaNet
mAP@0.50: 75.9
mAP@0.5:0.95: 48.9
object-detection-on-sfchdVFNet+SCALE
mAP@0.50: 76.6
mAP@0.5:0.95: 51.4
object-detection-on-sfchdYOLOv8
mAP@0.50: 77.9
mAP@0.5:0.95: 52.2
object-detection-on-sfchdYOLOv5
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.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Large, Complex, and Realistic Safety Clothing and Helmet Detection: Dataset and Method | Papers | HyperAI