dronefreak commited on
Commit
60dbbd9
·
verified ·
1 Parent(s): ae0e2f0

Upload model artifacts

Browse files
.gitattributes CHANGED
@@ -33,3 +33,9 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ BoxF1_curve.png filter=lfs diff=lfs merge=lfs -text
37
+ BoxP_curve.png filter=lfs diff=lfs merge=lfs -text
38
+ BoxPR_curve.png filter=lfs diff=lfs merge=lfs -text
39
+ BoxR_curve.png filter=lfs diff=lfs merge=lfs -text
40
+ confusion_matrix.png filter=lfs diff=lfs merge=lfs -text
41
+ visdrone_showcase.gif filter=lfs diff=lfs merge=lfs -text
BoxF1_curve.png ADDED

Git LFS Details

  • SHA256: 31de3893ff71ed97b9e6b79f558fd73c3726ebec7bc21c08cf475b11da86da16
  • Pointer size: 131 Bytes
  • Size of remote file: 346 kB
BoxPR_curve.png ADDED

Git LFS Details

  • SHA256: 25d07d792b3b750ab15ae05e7a726dff725a1d533747731ed256500ffc12a16b
  • Pointer size: 131 Bytes
  • Size of remote file: 337 kB
BoxP_curve.png ADDED

Git LFS Details

  • SHA256: 9df51a660b7cb1b09927d27eb7799497478d0299f9723571e4f7f5551854dda1
  • Pointer size: 131 Bytes
  • Size of remote file: 400 kB
BoxR_curve.png ADDED

Git LFS Details

  • SHA256: 5a79dd3228909a33010eb776eccbfbd0aad55b3754624f191b7e4ca738ae2ab6
  • Pointer size: 131 Bytes
  • Size of remote file: 328 kB
README.md ADDED
@@ -0,0 +1,299 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+
4
+ pipeline_tag: object-detection
5
+
6
+ library_name: ultralytics
7
+
8
+ datasets:
9
+ - Voxel51/VisDrone2019-DET
10
+
11
+ tags:
12
+ - object-detection
13
+ - aerial-imagery
14
+ - drone
15
+ - visdrone
16
+ - ultralytics
17
+ - pytorch
18
+ - computer-vision
19
+
20
+ metrics:
21
+ - map50
22
+ - map50-95
23
+ - precision
24
+ - recall
25
+ - f1
26
+
27
+ base_model: "Ultralytics/YOLO11"
28
+ ---
29
+
30
+
31
+ # YOLOv11l Finetuned on VisDrone
32
+
33
+ Fine-tuned YOLOv11l object detector for aerial imagery using the VisDrone benchmark dataset.
34
+
35
+ This model is part of the **VisDrone Detection Model Zoo**, a collection of YOLO models trained and evaluated under a common pipeline for aerial object detection.
36
+
37
+ ## Detection Showcase
38
+
39
+ <p align="center">
40
+ <img src="visdrone_showcase.gif" alt="VisDrone Detection Demo">
41
+ </p>
42
+
43
+ ---
44
+
45
+ ## Performance
46
+
47
+ | Metric | Score (%) |
48
+ | ---------- | --------------- |
49
+ | mAP@50 | 37.14 |
50
+ | mAP@50-95 | 21.85 |
51
+ | Precision | 51.87 |
52
+ | Recall | 40.33 |
53
+ | F1 Score | 45.38 |
54
+ | Parameters | - |
55
+ | FLOPs | - |
56
+
57
+ ---
58
+
59
+ ## Evaluation Protocol
60
+
61
+ Metrics reported in this model card are computed on the VisDrone test set with ground-truth annotations available for evaluation.
62
+
63
+ ---
64
+
65
+ ## VisDrone Model Zoo
66
+
67
+ | Rank | Model | mAP@50 | mAP@50-95 | Precision | Recall |
68
+ | -------------------------- | -------------------- | ------------- | --------------- | ----------------- | -------------- |
69
+ | | | | | | |
70
+ | 1 | YOLOv9e | 40.02 | 23.73 | 54.78 | 42.42 |
71
+ | | | | | | |
72
+ | 2 | YOLOv11x | 38.44 | 22.6 | 52.41 | 41.43 |
73
+ | | | | | | |
74
+ | 3 | YOLOv26x | 38.33 | 22.48 | 52.91 | 41.06 |
75
+ | | | | | | |
76
+ | 4 | YOLOv11l | 37.14 | 21.85 | 51.87 | 40.33 |
77
+ | | | | | | |
78
+ | 5 | YOLOv10x | 37.24 | 21.81 | 52.59 | 39.84 |
79
+ | | | | | | |
80
+ | 6 | YOLOv26l | 37.65 | 21.75 | 51.6 | 40.42 |
81
+ | | | | | | |
82
+ | 7 | YOLOv9c | 37.22 | 21.73 | 51.99 | 39.77 |
83
+ | | | | | | |
84
+ | 8 | YOLOv8x | 36.81 | 21.52 | 51.91 | 39.78 |
85
+ | | | | | | |
86
+ | 9 | YOLOv10l | 35.95 | 21.09 | 52.13 | 38.48 |
87
+ | | | | | | |
88
+ | 10 | YOLOv9m | 36.19 | 20.95 | 51.05 | 39.12 |
89
+ | | | | | | |
90
+ | 11 | YOLOv8m | 34.39 | 19.95 | 48.18 | 38.2 |
91
+ | | | | | | |
92
+ | 12 | YOLOv8s | 31.95 | 18.24 | 45.99 | 35.49 |
93
+ | | | | | | |
94
+ | 13 | YOLOv8n | 28.18 | 15.77 | 40.86 | 31.81 |
95
+ | | | | | | |
96
+ | 14 | YOLOv11n | 27.59 | 15.46 | 39.58 | 31.74 |
97
+ | | | | | | |
98
+ | 15 | YOLOv10n | 27.65 | 15.32 | 41.02 | 31.68 |
99
+ | | | | | | |
100
+ | 16 | YOLOv26n | 26.73 | 14.64 | 38.6 | 31.14 |
101
+ | | | | | | |
102
+
103
+ ---
104
+
105
+ ## Per-Class Performance
106
+
107
+ | Class | mAP@50 | mAP@50-95 |
108
+ | -------------------------- | --------------- | ----------------- |
109
+ | | | |
110
+ | pedestrian | 34.61 | 14.1 |
111
+ | | | |
112
+ | people | 17.67 | 6.19 |
113
+ | | | |
114
+ | bicycle | 14.24 | 6.22 |
115
+ | | | |
116
+ | car | 75.96 | 49.02 |
117
+ | | | |
118
+ | van | 41.02 | 27.83 |
119
+ | | | |
120
+ | truck | 48.34 | 32.22 |
121
+ | | | |
122
+ | tricycle | 23.48 | 12.93 |
123
+ | | | |
124
+ | awning-tricycle | 19.24 | 11.52 |
125
+ | | | |
126
+ | bus | 59.48 | 43.13 |
127
+ | | | |
128
+ | motor | 37.39 | 15.37 |
129
+ | | | |
130
+
131
+ ---
132
+
133
+ ## Evaluation Visualizations
134
+
135
+ ### Precision-Recall Curve
136
+
137
+ ![PR Curve](BoxPR_curve.png)
138
+
139
+ ### F1 Curve
140
+
141
+ ![F1 Curve](BoxF1_curve.png)
142
+
143
+ ### Confusion Matrix
144
+
145
+ ![Confusion Matrix](confusion_matrix.png)
146
+
147
+ ---
148
+
149
+ ## Dataset
150
+
151
+ VisDrone is a large-scale benchmark for object detection in aerial imagery captured from unmanned aerial vehicles (UAVs).
152
+
153
+ The dataset contains diverse scenes including:
154
+
155
+ * Urban environments
156
+ * Residential areas
157
+ * Traffic intersections
158
+ * Crowded pedestrian regions
159
+
160
+ ### Classes
161
+
162
+ * pedestrian
163
+ * people
164
+ * bicycle
165
+ * car
166
+ * van
167
+ * truck
168
+ * tricycle
169
+ * awning-tricycle
170
+ * bus
171
+ * motor
172
+
173
+ ---
174
+
175
+ ## Usage
176
+
177
+ ### Install Dependencies
178
+
179
+ ```bash
180
+ pip install ultralytics huggingface_hub
181
+ ```
182
+
183
+ ### Load Model from Hugging Face
184
+
185
+ ```python
186
+ from huggingface_hub import hf_hub_download
187
+ from ultralytics import YOLO
188
+
189
+ weights = hf_hub_download(
190
+ repo_id="dronefreak/yolov11l-visdrone",
191
+ filename="best.pt"
192
+ )
193
+
194
+ model = YOLO(weights)
195
+ ```
196
+
197
+ ### Run Inference
198
+
199
+ ```python
200
+ results = model.predict(
201
+ source="image.jpg",
202
+ conf=0.25
203
+ )
204
+
205
+ results[0].show()
206
+ ```
207
+
208
+ ---
209
+
210
+ ## Training Configuration
211
+
212
+ | Setting | Value |
213
+ | ---------------- | ------------------------------- |
214
+ | Epochs | 300 |
215
+ | Dataset | VisDrone2019-DET |
216
+ | Framework | Ultralytics YOLO |
217
+ | Training Toolkit | VisDrone Dataset Python Toolkit |
218
+
219
+ ---
220
+
221
+ ## Repository Contents
222
+
223
+ ```text
224
+ best.pt
225
+ results.csv
226
+ args.yaml
227
+ BoxPR_curve.png
228
+ BoxF1_curve.png
229
+ confusion_matrix.png
230
+ assets/visdrone_showcase.gif
231
+ README.md
232
+ ```
233
+
234
+ ---
235
+
236
+ ## Related Resources
237
+
238
+ * VisDrone Detection Model Zoo (Hugging Face Collection)
239
+ * VisDrone Dataset Python Toolkit: https://github.com/dronefreak/VisDrone-dataset-python-toolkit
240
+ * VisDrone Dataset: https://github.com/VisDrone/VisDrone-Dataset
241
+
242
+ ---
243
+
244
+ ## Training Framework
245
+
246
+ These models were trained using the VisDrone Dataset Python Toolkit, an open-source framework for aerial object detection research and benchmarking on the VisDrone dataset.
247
+
248
+ Features include:
249
+
250
+ * Dataset preparation and conversion utilities
251
+ * Training and evaluation pipelines
252
+ * Detection benchmarking
253
+ * Visualization tools
254
+ * Support for multiple YOLO model families
255
+
256
+ Repository:
257
+
258
+ https://github.com/dronefreak/VisDrone-dataset-python-toolkit
259
+
260
+ If you find these models useful, please consider starring the repository.
261
+
262
+ ---
263
+
264
+ ## Known Limitations
265
+
266
+ Performance may degrade in:
267
+
268
+ * Extremely dense crowds
269
+ * Heavy occlusions
270
+ * Severe motion blur
271
+ * Very small objects occupying only a few pixels
272
+ * Night-time or low-light aerial imagery
273
+
274
+ ---
275
+
276
+ ## Citation
277
+
278
+ If you use this model in your research, please consider citing:
279
+
280
+ 1. The VisDrone dataset
281
+ 2. The original YOLO architecture
282
+ 3. The VisDrone Detection Toolkit
283
+
284
+ ```bibtex
285
+ @article{visdrone2019,
286
+ title={Vision Meets Drones: A Challenge},
287
+ author={Zhu, Pengfei and Wen, Longyin and Du, Dawei and Bian, Xiao and Ling, Haibin and Hu, Qinghua},
288
+ journal={International Journal of Computer Vision},
289
+ year={2021}
290
+ }
291
+
292
+ @software{Saksena_VisDrone_Detection_Toolkit_2025,
293
+ author = {Saksena, Saumya Kumaar},
294
+ title = {VisDrone Detection Toolkit: Modern PyTorch Implementation for Aerial Object Detection},
295
+ url = {https://github.com/dronefreak/VisDrone-dataset-python-toolkit},
296
+ version = {2.0.0},
297
+ year = {2025}
298
+ }
299
+ ```
best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:02588f0bb309ded934641f0fc40b8583a44e6a6bd5929833611a956fab96b4fe
3
+ size 51209177
confusion_matrix.png ADDED

Git LFS Details

  • SHA256: c94bb7282453b734f16d668e7013b6dbf76c8602edcee86f77c0486696e0a08f
  • Pointer size: 131 Bytes
  • Size of remote file: 299 kB
visdrone_showcase.gif ADDED

Git LFS Details

  • SHA256: 69e366100cb1fa9d8a75182295d723fa2cd347b4361cb06181ae1d7c9efaf405
  • Pointer size: 133 Bytes
  • Size of remote file: 45 MB