Helmet Detection Machine Learning Model - Computer Vision

This is a Jupyter Notebook that can detect whether someone is wearing a helmet or not. The program allows you to take an image to test with using your webcam

Getting Started

These instructions will get you a copy of the project up and running on your local machine for development and testing purposes. See deployment for notes on how to deploy the project on a live system.

Prerequisites

• Webcam needed to 'see' the helmet
• Jupyter Notebook to run the model

Deployment

This will run locally through Jupyter Notebooks. Alternatively, you could use Google Colab.

Built With

• PyTorch