https://github.com/srjchsv/icy-bridges-detection

Simple object detection using Yolov8
https://github.com/srjchsv/icy-bridges-detection

computer-vision cv data-science jupyter python pytorch ultralytics yolov8

Last synced: 2 months ago
JSON representation

Simple object detection using Yolov8

• Host: GitHub
• URL: https://github.com/srjchsv/icy-bridges-detection
• Owner: srjchsv
• Created: 2023-09-03T12:20:56.000Z (almost 3 years ago)
• Default Branch: main
• Last Pushed: 2023-09-03T13:13:38.000Z (almost 3 years ago)
• Last Synced: 2025-03-21T14:19:12.649Z (over 1 year ago)
• Topics: computer-vision, cv, data-science, jupyter, python, pytorch, ultralytics, yolov8
• Language: Jupyter Notebook
• Homepage:
• Size: 14.8 MB
• Stars: 0
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: readme.md

Awesome Lists containing this project

README

          
# Object detection project

Task is to detect objects - icy bridges, on images coming from road cameras.

- Data from ["rtp-aws"](https://github.com/rtp-aws/devpost_aws_disaster_response/tree/main/icy-bridge), I also tried to solve this problem.

- Made annotattion with ["CVAT.ai"](https://www.cvat.ai/) with free account

  - Created 2 classes:

    - bridges

    - icy_bridges

- Split data into train and validate batches

- Trained Yolo v8 image detection model

- Validated the trained model

Its a simple solution to try solve the problem and learn the CV technologies and object detection. 

## Here is what it was able to detect:

### Labeled objects

!["How it should be"](/runs/detect/val2/val_batch0_labels.jpg)

### Whats model detected on validation

!["What model detected"](/runs/detect/val2/val_batch0_pred.jpg)

### Confusion Matrix

!["Confusion Martix"](/runs/detect/val2/confusion_matrix_normalized.png)

## useful resources:

Train Yolov8 object detection on a custom dataset [https://www.youtube.com/watch?v=m9fH9OWn8YM](https://www.youtube.com/watch?v=m9fH9OWn8YM)