Helmet Detection - YOLOv5

Jul 29, 2022

Grid of construction site photos with YOLOv5 helmet detection bounding boxes

Detail Project

Deskripsi

Pada project ini dibuat sebuah model deteksi helm keselamatan dengan menggunakan teknologi YOLOv5. Dataset yang digunakan didapatkan dari website Kaggle. Dataset berisi 5000 gambar yang sudah dianotasi. Dataset memiliki 3 kelas yaitu Helmet, Person, dan Head. Namun, pada project ini kelas yang digunakan hanya Helmet dan Head saja. Project ini merupakan project tugas besar pada matakuliah Computer Vision.

Berikut adalah hasil kuantitatif yang didapatkan pada proses pengujian:

MetrikValue
Precision94%
Recall89.3%
F1-Score92%
[email protected]95%
[email protected]:.9562.6%

Untuk hasil kuantitatif maupun kualitatif yang lebih detail dapat dilihat pada link berikut: W&B Report

Baskoro Adi