This project focuses on building a deep learning–based computer vision system to automatically detect whether workers are wearing safety helmets in industrial or construction environments. The solution improves workplace safety monitoring by automating compliance checks and reducing reliance on manual supervision.
The primary goal was to develop an image classification model capable of distinguishing between workers with and without helmets. Such a system enhances safety enforcement, reduces accident risks, and supports real-time safety monitoring at scale.
- Source: Provided as part of the project coursework
- Size: 631 labeled images
- Categories:
With Helmet– Workers wearing helmetsWithout Helmet– Workers without helmets
- Data Preprocessing – Converted images to grayscale, normalized pixel values, and split data into training, validation, and test sets.
- Model Development – Built and trained CNN-based classifiers, including a baseline CNN and transfer learning models (VGG-16).
- Model Enhancement – Applied data augmentation, fine-tuned architectures, and compared model performances.
- Evaluation & Insights – Selected the best-performing model for deployment in real-world safety applications.
- Delivered a highly accurate model for helmet detection across diverse real-world conditions.
- Enabled scalable, automated safety compliance monitoring.
- Demonstrated the potential of computer vision in workplace safety and industrial automation.
- Language: Python
- Libraries: TensorFlow/Keras, OpenCV, NumPy, Matplotlib, Seaborn
- Tools: Jupyter Notebook / Google Colab
Sandesh S. Badwaik