Recently, while exploring computer vision got a chance to train YoloV3 on custom dataset for object detection.

We custom trained the YOLO V3 to detect following classes:
– hardhat
– vest
– mask
– boots

Below is a short video demonstrating how amazingly the model is able to detect these objects.

Code for this can be found here.