srjchsv / icy-bridges-detection

Simple object detection using Yolov8

Object detection project

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

• Data from "rtp-aws", I also tried to solve this problem.
• Made annotattion with "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

Whats model detected on validation

Confusion Matrix

useful resources:

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