Inspiration

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 https://www.kaggle.com/vodan37/yolo-helmethead

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

• python
• yolo