Zarinaaa commited on
Commit
66a0370
·
verified ·
1 Parent(s): 9e57b83

Create README

Browse files
Files changed (1) hide show
  1. README.md +128 -0
README.md ADDED
@@ -0,0 +1,128 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - yolov8
5
+ - object-detection
6
+ - railway
7
+ - wagon-number-recognition
8
+ - ocr
9
+ - computer-vision
10
+ - ultralytics
11
+ datasets:
12
+ - custom
13
+ language:
14
+ - en
15
+ pipeline_tag: object-detection
16
+ ---
17
+
18
+ # YOLOv8 — Railway Wagon Big Number Detection
19
+
20
+ <p align="center">
21
+ <img src="001_20241208064334_[M][0@0][0].jpg" alt="Example prediction" width="600"/>
22
+ </p>
23
+
24
+ ## Model Description
25
+
26
+ A fine-tuned **YOLOv8** model for detecting and localizing **large identification numbers** on railway wagons. This model was developed as part of a **government railway monitoring system** to automate wagon identification in real-time video streams.
27
+
28
+ The model accurately detects oversized wagon numbers painted on the side of freight and passenger cars, even under challenging conditions such as varying lighting, weather, partial occlusion, and motion blur.
29
+
30
+ ## Intended Use
31
+
32
+ | Use Case | Description |
33
+ |----------|-------------|
34
+ | **Railway monitoring** | Automated wagon identification in surveillance camera feeds |
35
+ | **Logistics tracking** | Real-time wagon number extraction for freight management |
36
+ | **Infrastructure inspection** | Integration into railway digitization pipelines |
37
+
38
+ > ⚠️ **Not intended for:** General-purpose OCR, license plate recognition, or non-railway number detection.
39
+
40
+ ## Performance
41
+
42
+ | Metric | Score |
43
+ |--------|-------|
44
+ | Precision | `XX.X%` |
45
+ | Recall | `XX.X%` |
46
+ | F1 Score | `XX.X%` |
47
+ | Accuracy | `XX.X%` |
48
+
49
+ <!-- 🔧 TODO: Replace XX.X% with your actual metric values -->
50
+
51
+ ## Training Details
52
+
53
+ ### Dataset
54
+
55
+ - **Source:** Custom-annotated dataset of railway wagon images
56
+ - **Size:** `X,XXX` images <!-- TODO: fill in -->
57
+ - **Annotation format:** YOLO (bounding boxes)
58
+ - **Classes:** 1 (big_number)
59
+ - **Train/Val split:** 80/20
60
+
61
+ ### Hyperparameters
62
+
63
+ | Parameter | Value |
64
+ |-----------|-------|
65
+ | Base model | YOLOv8n / YOLOv8s <!-- specify which --> |
66
+ | Image size | 640 |
67
+ | Batch size | 16 |
68
+ | Epochs | `XXX` |
69
+ | Optimizer | AdamW |
70
+ | Learning rate | 0.01 |
71
+ | Device | NVIDIA GPU |
72
+
73
+ <!-- 🔧 TODO: Update hyperparameters with actual values -->
74
+
75
+ ### Training Framework
76
+
77
+ - [Ultralytics YOLOv8](https://github.com/ultralytics/ultralytics)
78
+ - Python 3.10+
79
+ - PyTorch 2.x
80
+
81
+ ## How to Use
82
+
83
+ ```python
84
+ from ultralytics import YOLO
85
+
86
+ # Load the model
87
+ model = YOLO("best.pt")
88
+
89
+ # Run inference on an image
90
+ results = model("wagon_image.jpg")
91
+
92
+ # Display results
93
+ for result in results:
94
+ boxes = result.boxes
95
+ for box in boxes:
96
+ print(f"Big Number detected | Confidence: {box.conf[0]:.2f} | BBox: {box.xyxy[0]}")
97
+ ```
98
+
99
+ ## Model Files
100
+
101
+ | File | Description |
102
+ |------|-------------|
103
+ | `best.pt` | Best checkpoint (highest validation metric) |
104
+ | `last.pt` | Last training epoch checkpoint |
105
+
106
+ ## Limitations
107
+
108
+ - Trained specifically on railway wagon numbers; may not generalize to other number detection tasks
109
+ - Performance may degrade on heavily occluded or damaged wagon surfaces
110
+ - Optimized for daytime footage; low-light performance may vary
111
+
112
+ ## Citation
113
+
114
+ ```bibtex
115
+ @misc{yolo8_bignumbers_2024,
116
+ author = {Zarina},
117
+ title = {YOLOv8 Railway Wagon Big Number Detection},
118
+ year = {2024},
119
+ url = {https://huggingface.co/Zarinaaa/yolo8_BigNumbers_model}
120
+ }
121
+ ```
122
+
123
+ ## Author
124
+
125
+ **Zarina** — Machine Learning Engineer specializing in Computer Vision, NLP, and Speech Technologies.
126
+
127
+ - 🤗 [HuggingFace](https://huggingface.co/Zarinaaa)
128
+ - 💼 [LinkedIn](https://linkedin.com/in/YOUR_LINKEDIN) <!-- TODO: add your link -->