https://github.com/viddesh1/helmet_test_1

Implementation of YOLOv8 on custom dataset to detect "bike rider", "helmet" and "no helmet"
https://github.com/viddesh1/helmet_test_1

jupyter-notebook python3 ultralytics yolov8

Last synced: 7 days ago
JSON representation

Implementation of YOLOv8 on custom dataset to detect "bike rider", "helmet" and "no helmet"

• Host: GitHub
• URL: https://github.com/viddesh1/helmet_test_1
• Owner: Viddesh1
• Created: 2023-10-09T08:19:42.000Z (about 1 year ago)
• Default Branch: main
• Last Pushed: 2024-06-18T13:46:08.000Z (5 months ago)
• Last Synced: 2024-06-18T16:31:18.180Z (5 months ago)
• Topics: jupyter-notebook, python3, ultralytics, yolov8
• Language: Jupyter Notebook
• Homepage:
• Size: 12.4 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# Bike Rider Helmet Detection using YOLOv8

This repository contains a walk through of implementation of bike rider helmet detection using YOLOv8. As we know that bike riders who do not wear helmet may which result in fatal accidents and death in some cases. The goal is to create a ML/DL model that can detect if a person is wearing helment or not. May sure to select GPU for running through the notebook.

### Methodology :-

1) Data Collection:- Collect images/videos of bike rider, helmet and no helmet for the model to train upon.

2) Data pre-processing:- Pre-processing and autolabel images and videos using foundation model like DINO and SAM (Segment anything model).

3) Train YOLOv8 Model

4) Evaluate Target Model

5) Run Inference on images and videos

## This repository output may change in the near future:-

https://drive.google.com/drive/folders/1M4FckJJeyPQTTWqgo6xWhW8L4tf0EJ4l?usp=sharing

# Also see

1) https://github.com/Viddesh1/Bike-Helmet-Detection

2) https://github.com/Viddesh1/Bike-Helmet-Detectionv2

3) https://github.com/Viddesh1/Bike-Helmet-Detection-Docs