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.
- Python
- YOLOv8 Classification
- Ultralytics
- OpenCV
- NumPy
- Pandas
- Matplotlib
- Seaborn
- Scikit-learn
- Google Colab (GPU)
Weather Dataset
https://data.mendeley.com/datasets/4drtyfjtfy/1
Dataset Classes:
- cloudy
- rain
- shine
- sunrise
- 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