DOI: 10.3390/buildings16122447 ISSN: 2075-5309

Reliable Vision-Based PPE Detection for Construction Safety in Adverse Environmental Conditions

Sujan Gyawali, Ali Mohammadjafari, Saurav Ghimire, Mahmoud Habibnezhad

Adverse imaging conditions such as fog, rain, and low light degrade the reliability of vision-based Personal Protective Equipment (PPE) detection systems on construction sites, yet most existing models are trained under clear-weather assumptions. This paper introduces a physics-based weather augmentation framework integrated with the YOLOv8n architecture to improve PPE detection robustness under adverse environmental conditions. The original Color Helmet and Vest (CHV) dataset was expanded from 1330 clear-weather images to 6650 images across five conditions using four physically grounded augmentation models: the Koschmieder atmospheric scattering model for fog, the Garg–Nayar streak model for rain, gamma-corrected attenuation with Poisson–Gaussian noise for low light, and a PSF-based glare model for bright sunlight. The weather-resistant model, a clear-weather baseline, and an augmented baseline were evaluated on the same 665-image weather-augmented test set. The weather-resistant model achieves 89.2% mAP50, a 5.7 percentage-point improvement over the clear-weather baseline (83.5%), with a nearly four-fold improvement in cross-condition stability (standard deviation 1.5% vs. 5.7%). Under matched training-data volume, the weather-resistant model still outperforms a conventionally augmented baseline across all five simulated conditions, indicating that these gains stem from physics-based modeling rather than larger training-data volume. The largest gain occurs under low light, where mAP50 improves from 73.4% to 87.9%. Gradient-weighted Class Activation Mapping (Grad-CAM) analysis confirms that the weather-resistant model directs more attention toward PPE regions across all conditions, with the largest improvement under low light (+10.0 percentage points). The lightweight design (3.0 M parameters) and quantitative and qualitative validation on 205 annotated real-world construction site images under normal and low-light conditions provide preliminary evidence of practical applicability.

More from our Archive