update
Browse files
README.md
CHANGED
|
@@ -41,10 +41,8 @@ The model accurately detects oversized wagon numbers painted on the side of frei
|
|
| 41 |
|
| 42 |
| Metric | Score |
|
| 43 |
|--------|-------|
|
| 44 |
-
|
|
| 45 |
-
|
| 46 |
-
| F1 Score | `XX.X%` |
|
| 47 |
-
| Accuracy | `XX.X%` |
|
| 48 |
|
| 49 |
<!-- 🔧 TODO: Replace XX.X% with your actual metric values -->
|
| 50 |
|
|
@@ -53,7 +51,7 @@ The model accurately detects oversized wagon numbers painted on the side of frei
|
|
| 53 |
### Dataset
|
| 54 |
|
| 55 |
- **Source:** Custom-annotated dataset of railway wagon images
|
| 56 |
-
- **Size:**
|
| 57 |
- **Annotation format:** YOLO (bounding boxes)
|
| 58 |
- **Classes:** 1 (big_number)
|
| 59 |
- **Train/Val split:** 80/20
|
|
@@ -65,7 +63,7 @@ The model accurately detects oversized wagon numbers painted on the side of frei
|
|
| 65 |
| Base model | YOLOv8n / YOLOv8s <!-- specify which --> |
|
| 66 |
| Image size | 640 |
|
| 67 |
| Batch size | 16 |
|
| 68 |
-
| Epochs | `
|
| 69 |
| Optimizer | AdamW |
|
| 70 |
| Learning rate | 0.01 |
|
| 71 |
| Device | NVIDIA GPU |
|
|
|
|
| 41 |
|
| 42 |
| Metric | Score |
|
| 43 |
|--------|-------|
|
| 44 |
+
| F1 Score | `98.1%` |
|
| 45 |
+
|
|
|
|
|
|
|
| 46 |
|
| 47 |
<!-- 🔧 TODO: Replace XX.X% with your actual metric values -->
|
| 48 |
|
|
|
|
| 51 |
### Dataset
|
| 52 |
|
| 53 |
- **Source:** Custom-annotated dataset of railway wagon images
|
| 54 |
+
- **Size:** 1240X800 images <!-- TODO: fill in -->
|
| 55 |
- **Annotation format:** YOLO (bounding boxes)
|
| 56 |
- **Classes:** 1 (big_number)
|
| 57 |
- **Train/Val split:** 80/20
|
|
|
|
| 63 |
| Base model | YOLOv8n / YOLOv8s <!-- specify which --> |
|
| 64 |
| Image size | 640 |
|
| 65 |
| Batch size | 16 |
|
| 66 |
+
| Epochs | `4` |
|
| 67 |
| Optimizer | AdamW |
|
| 68 |
| Learning rate | 0.01 |
|
| 69 |
| Device | NVIDIA GPU |
|