Skip to content

Repository files navigation

Weather Image Classification with YOLOv8

A mini Classification project focused on image classification using the YOLOv8 Classification model.
This project was created to practice train model workflow, dataset preparation, model training, and visualization using Google Colab with GPU acceleration.

image image confusion_matrix image

Tech Stack

  • Python
  • YOLOv8 Classification
  • Ultralytics
  • OpenCV
  • NumPy
  • Pandas
  • Matplotlib
  • Seaborn
  • Scikit-learn
  • Google Colab (GPU)

Dataset

Weather Dataset
https://data.mendeley.com/datasets/4drtyfjtfy/1

Dataset Classes:

  • cloudy
  • rain
  • shine
  • sunrise

Features

  • Automatic YOLO dataset structure preparation
  • Train / Validation split generation
  • Dataset imbalance visualization
  • YOLOv8 Classification training pipeline
  • GPU training support on Google Colab
  • Performance visualization:
    • Loss Curve
    • Accuracy Curve
    • Confusion Matrix
    • Validation Predictions
  • Automatic best epoch evaluation

About

Weather image classification project using YOLOv8 Classification with GPU training on Google Colab

Topics

Resources

Stars

Watchers

Forks

Releases

Packages

Contributors

Languages