Learn more call to action

Deteksi Helm Keselamatan Menggunakan YOLOv5

July 29, 2022
Deteksi Helm Keselamatan Menggunakan YOLOv5

Project Details

Tech Stack:Python, YOLOv5
Role:AI Programmer
Team:Baskoro Adi, Devani Paundri, Rahmat Tobi
Year:2022

Description

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:

Testing Result (All Classes)
Precision 94%
Recall 89.3%
F1-Score 92%
[email protected] 95%
[email protected]:.95 62.6%

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

Screenshot

Alt text
Beberapa hasil kualitatif dari model deteksi yang telah dibuat