🧠 Trash Detection Model (YOLOv11n)

This model detects and classifies trash types (plastic, paper, glass, trash) using YOLOv11n.
It was trained on the Trash Type Image Dataset from Kaggle.

📂 Dataset

Source: Kaggle Trash Type Image Dataset
Classes: plastic, paper, glass, trash
License: Follows the original dataset's Kaggle terms.
The dataset was split into train and val folders for YOLO format.

⚙️ Training Details

Base model: ultralytics/yolo11n.pt
Framework: Ultralytics YOLO
Image size: 320
Epochs: 30
Batch size: 2
Hardware: Jetson Orin Nano

🧩 Usage

from huggingface_hub import hf_hub_download
from ultralytics import YOLO

model_path = hf_hub_download(
    repo_id="Alope/trash-detection-yolo11n",
    filename="best.pt"
)
model = YOLO(model_path)
results = model.predict(source="test.jpg", show=True)

Downloads last month: -; Downloads are not tracked for this model. How to track