hajin4199 commited on
Commit
3e4c1cb
·
1 Parent(s): fd7ca9b
Files changed (3) hide show
  1. README.md +46 -0
  2. best.pt +3 -0
  3. data.yaml +2 -0
README.md ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: agpl-3.0
3
+ base_model: ultralytics/yolo11n
4
+ dataset: farzadnekouei/trash-type-image-dataset
5
+ tags:
6
+ - object-detection
7
+ - yolov11
8
+ - trash-detection
9
+ ---
10
+
11
+ # 🧠 Trash Detection Model (YOLOv11n)
12
+
13
+ This model detects and classifies **trash types (plastic, paper, glass, trash)** using YOLOv11n.
14
+ It was trained on the [Trash Type Image Dataset](https://www.kaggle.com/datasets/farzadnekouei/trash-type-image-dataset) from Kaggle.
15
+
16
+ ---
17
+
18
+ ## 📂 Dataset
19
+ - **Source:** [Kaggle Trash Type Image Dataset](https://www.kaggle.com/datasets/farzadnekouei/trash-type-image-dataset)
20
+ - **Classes:** `plastic`, `paper`, `glass`, `trash`
21
+ - **License:** Follows the original dataset's Kaggle terms.
22
+ - The dataset was split into `train` and `val` folders for YOLO format.
23
+
24
+ ---
25
+
26
+ ## ⚙️ Training Details
27
+ - **Base model:** `ultralytics/yolo11n.pt`
28
+ - **Framework:** [Ultralytics YOLO](https://github.com/ultralytics/ultralytics)
29
+ - **Image size:** 320
30
+ - **Epochs:** 30
31
+ - **Batch size:** 2
32
+ - **Hardware:** Jetson Orin Nano
33
+
34
+ ---
35
+
36
+ ## 🧩 Usage
37
+ ```python
38
+ from huggingface_hub import hf_hub_download
39
+ from ultralytics import YOLO
40
+
41
+ model_path = hf_hub_download(
42
+ repo_id="Alope/trash-detection-yolo11n",
43
+ filename="best.pt"
44
+ )
45
+ model = YOLO(model_path)
46
+ results = model.predict(source="test.jpg", show=True)
best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13554af414247d33db7d56551e1afb2cb7d0ac5462de5d66df96a81a02fe648d
3
+ size 5425562
data.yaml ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ nc: 4
2
+ names: ['glass', 'paper', 'plastic', 'trash']