diazprasetyo/garbage_classification

Garbage Image Detection Model

This repository contains a YOLO-based deep learning model trained to classify garbage images. The model outputs a confidence level bar chart of predicted image.

🏆 Model Overview

• Model: YOLO-based image classifier
• Task: Garbage classification
• Output: Confidence bar chart with a single predicted class
• Framework: Ultralytics YOLO
• Dataset: https://www.kaggle.com/datasets/farzadnekouei/trash-type-image-dataset

📥 Download the Model

You can download the trained model (best.pt) using:

wget -O best.pt 'https://huggingface.co/diazprasetyo/garbage_classification/blob/main/best.pt'

Alternatively, you can load it directly in Python:

from ultralytics import YOLO
model = YOLO('best.pt')

🚀 How to Use

1️⃣ Install Dependencies

pip install ultralytics streamlit plotly pillow

2️⃣ Run the Streamlit App

streamlit run app.py

3️⃣ Upload an Image

• The app allows you to upload an image of garbage.
• The model predicts the type of garbage and displays a confidence level bar chart.

📄 License

This model is licensed under the MIT License. See LICENSE for details.

📢 Citation

If you use this model, please cite:

@misc{diazprasetyo2025,
  title={Garbage Image Detection Model},
  author={Diaz Prasetyo},
  year={2025},
  url={https://huggingface.co/diazprasetyo/garbage_classification}
}

🤝 Contributing

Feel free to open issues or pull requests to improve this model and its documentation.