Safety in Construction sites and in other workplaces is important.

What it does

Detect Hard Hat in videos, images (predict boundary boxes for head,hat )

Data set

Data set is downloaded from

How we built it

using yolov3 and improved accuracy in yolov4

Challenges we ran into

To train the model is the biggest problem for me as I only have CPU , but using google colab I am able to train my models.

What's next for worker Hat

To make a hardware model by deploying the model on Arduino

Built With

Share this project: