HardHead-Detection-for-Safety-Surveilance

HardHead Detection for Safety Surveilance

Problem Statement - sharepoint link

Task: Build A Personal Safety Equipment Detection System

Dataset Details:•

  • Hardhat/ head detection dataset is available here - Dataset Link

Dataset annotation format: Pascal VOC (XMLs)

Dataset Size:

  • Data with Annotations: 4,750
  • Data without Annotations (Test): 250

Collab Implemenetation

The entire Helmet/Head object detection implementation pipeline using YoloV3 can be found here colab_notebook

Sample Result

Sample Helmet and Head detection results can be found here.

Helmet detection Image1

Head detection - Here we can find that as in the dataset there are less head data as compared to helmet sets, so the model is not that good enough in finding heads in the 1st image. For this 2nd clean single person image head detection though is good.

Image2 Head2

Here also helmet detection result is quite good enough

Helmet_2

Note :

  • The project details and workings files can be found here - Drive Link
  • A guide to the project documentation can be found here

Reference: