Skip to content

YoloV8 model, trained for recognizing if construction workers are wearing their protection helmets in mandatory areas

Notifications You must be signed in to change notification settings

jomarkow/Safety-Helmet-Detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Safety Helmet Detection

Maintained

Index:

Description

Object detection system developed with deep learning techniques, capable to recognize if workers in construction areas are using their safety helmet in mandatory areas. The next objective it's to detect not only the helmets but also the people who puntually are unprotected, creating a warning.

Safety Helmet will be an integrated part of a dedicated ecosystem for secure safety habits on working dangerous areas, focusing on the scalability of the detection system trying to make it capable of covering the most scenarios posibles.

This project also haves a Colab repository for online testing at Colab page.

Made with:

Kaggle Python PyTorch Jupyter Notebook Colab

Technical

Dataset:

The dataset consist of 5000 images /data/images with annotations /data/labels, then divided on training and test subgroups.

Note

I used a converter /tools/coverter.py to addapt the previous labels format (.xml) to an accepted one (.txt), read script documentation before using

See full dataset documentation on this link.

Training:

Consist of a traditional YoloV8 training with a pretrained model, we apply a fine tunning for helmet detection using our custom dataset

from ultralytics import YOLO

model = YOLO("yolov8n.pt")
model.train(data="data.yaml", epochs=10)
result = model.val()
path = model.export(format="onnx")

Test output:

We recieve our trained model at /runs/detect/train6/weights/best.pt with some metrics for training analysis.

confusion-matrix-normalized.png

Colaborate

If you want to colaborate on this project, you are invited to create a Pull Request with a descriptive text of the changes/updates.

You can also contact me: jomarkow@gmail.com

We hope you liked the project, if so, I invite you to leave a star ⭐, thanks for read :).

About

YoloV8 model, trained for recognizing if construction workers are wearing their protection helmets in mandatory areas

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages