rfonod commited on
Commit
f264e2b
·
verified ·
1 Parent(s): 987f9d3

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,5 @@ 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
+ assets/PR_curve.png filter=lfs diff=lfs merge=lfs -text
37
+ assets/confusion_matrix_normalized.png filter=lfs diff=lfs merge=lfs -text
LICENSE ADDED
@@ -0,0 +1,395 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Attribution 4.0 International
2
+
3
+ =======================================================================
4
+
5
+ Creative Commons Corporation ("Creative Commons") is not a law firm and
6
+ does not provide legal services or legal advice. Distribution of
7
+ Creative Commons public licenses does not create a lawyer-client or
8
+ other relationship. Creative Commons makes its licenses and related
9
+ information available on an "as-is" basis. Creative Commons gives no
10
+ warranties regarding its licenses, any material licensed under their
11
+ terms and conditions, or any related information. Creative Commons
12
+ disclaims all liability for damages resulting from their use to the
13
+ fullest extent possible.
14
+
15
+ Using Creative Commons Public Licenses
16
+
17
+ Creative Commons public licenses provide a standard set of terms and
18
+ conditions that creators and other rights holders may use to share
19
+ original works of authorship and other material subject to copyright
20
+ and certain other rights specified in the public license below. The
21
+ following considerations are for informational purposes only, are not
22
+ exhaustive, and do not form part of our licenses.
23
+
24
+ Considerations for licensors: Our public licenses are
25
+ intended for use by those authorized to give the public
26
+ permission to use material in ways otherwise restricted by
27
+ copyright and certain other rights. Our licenses are
28
+ irrevocable. Licensors should read and understand the terms
29
+ and conditions of the license they choose before applying it.
30
+ Licensors should also secure all rights necessary before
31
+ applying our licenses so that the public can reuse the
32
+ material as expected. Licensors should clearly mark any
33
+ material not subject to the license. This includes other CC-
34
+ licensed material, or material used under an exception or
35
+ limitation to copyright. More considerations for licensors:
36
+ wiki.creativecommons.org/Considerations_for_licensors
37
+
38
+ Considerations for the public: By using one of our public
39
+ licenses, a licensor grants the public permission to use the
40
+ licensed material under specified terms and conditions. If
41
+ the licensor's permission is not necessary for any reason--for
42
+ example, because of any applicable exception or limitation to
43
+ copyright--then that use is not regulated by the license. Our
44
+ licenses grant only permissions under copyright and certain
45
+ other rights that a licensor has authority to grant. Use of
46
+ the licensed material may still be restricted for other
47
+ reasons, including because others have copyright or other
48
+ rights in the material. A licensor may make special requests,
49
+ such as asking that all changes be marked or described.
50
+ Although not required by our licenses, you are encouraged to
51
+ respect those requests where reasonable. More considerations
52
+ for the public:
53
+ wiki.creativecommons.org/Considerations_for_licensees
54
+
55
+ =======================================================================
56
+
57
+ Creative Commons Attribution 4.0 International Public License
58
+
59
+ By exercising the Licensed Rights (defined below), You accept and agree
60
+ to be bound by the terms and conditions of this Creative Commons
61
+ Attribution 4.0 International Public License ("Public License"). To the
62
+ extent this Public License may be interpreted as a contract, You are
63
+ granted the Licensed Rights in consideration of Your acceptance of
64
+ these terms and conditions, and the Licensor grants You such rights in
65
+ consideration of benefits the Licensor receives from making the
66
+ Licensed Material available under these terms and conditions.
67
+
68
+
69
+ Section 1 -- Definitions.
70
+
71
+ a. Adapted Material means material subject to Copyright and Similar
72
+ Rights that is derived from or based upon the Licensed Material
73
+ and in which the Licensed Material is translated, altered,
74
+ arranged, transformed, or otherwise modified in a manner requiring
75
+ permission under the Copyright and Similar Rights held by the
76
+ Licensor. For purposes of this Public License, where the Licensed
77
+ Material is a musical work, performance, or sound recording,
78
+ Adapted Material is always produced where the Licensed Material is
79
+ synched in timed relation with a moving image.
80
+
81
+ b. Adapter's License means the license You apply to Your Copyright
82
+ and Similar Rights in Your contributions to Adapted Material in
83
+ accordance with the terms and conditions of this Public License.
84
+
85
+ c. Copyright and Similar Rights means copyright and/or similar rights
86
+ closely related to copyright including, without limitation,
87
+ performance, broadcast, sound recording, and Sui Generis Database
88
+ Rights, without regard to how the rights are labeled or
89
+ categorized. For purposes of this Public License, the rights
90
+ specified in Section 2(b)(1)-(2) are not Copyright and Similar
91
+ Rights.
92
+
93
+ d. Effective Technological Measures means those measures that, in the
94
+ absence of proper authority, may not be circumvented under laws
95
+ fulfilling obligations under Article 11 of the WIPO Copyright
96
+ Treaty adopted on December 20, 1996, and/or similar international
97
+ agreements.
98
+
99
+ e. Exceptions and Limitations means fair use, fair dealing, and/or
100
+ any other exception or limitation to Copyright and Similar Rights
101
+ that applies to Your use of the Licensed Material.
102
+
103
+ f. Licensed Material means the artistic or literary work, database,
104
+ or other material to which the Licensor applied this Public
105
+ License.
106
+
107
+ g. Licensed Rights means the rights granted to You subject to the
108
+ terms and conditions of this Public License, which are limited to
109
+ all Copyright and Similar Rights that apply to Your use of the
110
+ Licensed Material and that the Licensor has authority to license.
111
+
112
+ h. Licensor means the individual(s) or entity(ies) granting rights
113
+ under this Public License.
114
+
115
+ i. Share means to provide material to the public by any means or
116
+ process that requires permission under the Licensed Rights, such
117
+ as reproduction, public display, public performance, distribution,
118
+ dissemination, communication, or importation, and to make material
119
+ available to the public including in ways that members of the
120
+ public may access the material from a place and at a time
121
+ individually chosen by them.
122
+
123
+ j. Sui Generis Database Rights means rights other than copyright
124
+ resulting from Directive 96/9/EC of the European Parliament and of
125
+ the Council of 11 March 1996 on the legal protection of databases,
126
+ as amended and/or succeeded, as well as other essentially
127
+ equivalent rights anywhere in the world.
128
+
129
+ k. You means the individual or entity exercising the Licensed Rights
130
+ under this Public License. Your has a corresponding meaning.
131
+
132
+
133
+ Section 2 -- Scope.
134
+
135
+ a. License grant.
136
+
137
+ 1. Subject to the terms and conditions of this Public License,
138
+ the Licensor hereby grants You a worldwide, royalty-free,
139
+ non-sublicensable, non-exclusive, irrevocable license to
140
+ exercise the Licensed Rights in the Licensed Material to:
141
+
142
+ a. reproduce and Share the Licensed Material, in whole or
143
+ in part; and
144
+
145
+ b. produce, reproduce, and Share Adapted Material.
146
+
147
+ 2. Exceptions and Limitations. For the avoidance of doubt, where
148
+ Exceptions and Limitations apply to Your use, this Public
149
+ License does not apply, and You do not need to comply with
150
+ its terms and conditions.
151
+
152
+ 3. Term. The term of this Public License is specified in Section
153
+ 6(a).
154
+
155
+ 4. Media and formats; technical modifications allowed. The
156
+ Licensor authorizes You to exercise the Licensed Rights in
157
+ all media and formats whether now known or hereafter created,
158
+ and to make technical modifications necessary to do so. The
159
+ Licensor waives and/or agrees not to assert any right or
160
+ authority to forbid You from making technical modifications
161
+ necessary to exercise the Licensed Rights, including
162
+ technical modifications necessary to circumvent Effective
163
+ Technological Measures. For purposes of this Public License,
164
+ simply making modifications authorized by this Section 2(a)
165
+ (4) never produces Adapted Material.
166
+
167
+ 5. Downstream recipients.
168
+
169
+ a. Offer from the Licensor -- Licensed Material. Every
170
+ recipient of the Licensed Material automatically
171
+ receives an offer from the Licensor to exercise the
172
+ Licensed Rights under the terms and conditions of this
173
+ Public License.
174
+
175
+ b. No downstream restrictions. You may not offer or impose
176
+ any additional or different terms or conditions on, or
177
+ apply any Effective Technological Measures to, the
178
+ Licensed Material if doing so restricts exercise of the
179
+ Licensed Rights by any recipient of the Licensed
180
+ Material.
181
+
182
+ 6. No endorsement. Nothing in this Public License constitutes or
183
+ may be construed as permission to assert or imply that You
184
+ are, or that Your use of the Licensed Material is, connected
185
+ with, or sponsored, endorsed, or granted official status by,
186
+ the Licensor or others designated to receive attribution as
187
+ provided in Section 3(a)(1)(A)(i).
188
+
189
+ b. Other rights.
190
+
191
+ 1. Moral rights, such as the right of integrity, are not
192
+ licensed under this Public License, nor are publicity,
193
+ privacy, and/or other similar personality rights; however, to
194
+ the extent possible, the Licensor waives and/or agrees not to
195
+ assert any such rights held by the Licensor to the limited
196
+ extent necessary to allow You to exercise the Licensed
197
+ Rights, but not otherwise.
198
+
199
+ 2. Patent and trademark rights are not licensed under this
200
+ Public License.
201
+
202
+ 3. To the extent possible, the Licensor waives any right to
203
+ collect royalties from You for the exercise of the Licensed
204
+ Rights, whether directly or through a collecting society
205
+ under any voluntary or waivable statutory or compulsory
206
+ licensing scheme. In all other cases the Licensor expressly
207
+ reserves any right to collect such royalties.
208
+
209
+
210
+ Section 3 -- License Conditions.
211
+
212
+ Your exercise of the Licensed Rights is expressly made subject to the
213
+ following conditions.
214
+
215
+ a. Attribution.
216
+
217
+ 1. If You Share the Licensed Material (including in modified
218
+ form), You must:
219
+
220
+ a. retain the following if it is supplied by the Licensor
221
+ with the Licensed Material:
222
+
223
+ i. identification of the creator(s) of the Licensed
224
+ Material and any others designated to receive
225
+ attribution, in any reasonable manner requested by
226
+ the Licensor (including by pseudonym if
227
+ designated);
228
+
229
+ ii. a copyright notice;
230
+
231
+ iii. a notice that refers to this Public License;
232
+
233
+ iv. a notice that refers to the disclaimer of
234
+ warranties;
235
+
236
+ v. a URI or hyperlink to the Licensed Material to the
237
+ extent reasonably practicable;
238
+
239
+ b. indicate if You modified the Licensed Material and
240
+ retain an indication of any previous modifications; and
241
+
242
+ c. indicate the Licensed Material is licensed under this
243
+ Public License, and include the text of, or the URI or
244
+ hyperlink to, this Public License.
245
+
246
+ 2. You may satisfy the conditions in Section 3(a)(1) in any
247
+ reasonable manner based on the medium, means, and context in
248
+ which You Share the Licensed Material. For example, it may be
249
+ reasonable to satisfy the conditions by providing a URI or
250
+ hyperlink to a resource that includes the required
251
+ information.
252
+
253
+ 3. If requested by the Licensor, You must remove any of the
254
+ information required by Section 3(a)(1)(A) to the extent
255
+ reasonably practicable.
256
+
257
+ 4. If You Share Adapted Material You produce, the Adapter's
258
+ License You apply must not prevent recipients of the Adapted
259
+ Material from complying with this Public License.
260
+
261
+
262
+ Section 4 -- Sui Generis Database Rights.
263
+
264
+ Where the Licensed Rights include Sui Generis Database Rights that
265
+ apply to Your use of the Licensed Material:
266
+
267
+ a. for the avoidance of doubt, Section 2(a)(1) grants You the right
268
+ to extract, reuse, reproduce, and Share all or a substantial
269
+ portion of the contents of the database;
270
+
271
+ b. if You include all or a substantial portion of the database
272
+ contents in a database in which You have Sui Generis Database
273
+ Rights, then the database in which You have Sui Generis Database
274
+ Rights (but not its individual contents) is Adapted Material; and
275
+
276
+ c. You must comply with the conditions in Section 3(a) if You Share
277
+ all or a substantial portion of the contents of the database.
278
+
279
+ For the avoidance of doubt, this Section 4 supplements and does not
280
+ replace Your obligations under this Public License where the Licensed
281
+ Rights include other Copyright and Similar Rights.
282
+
283
+
284
+ Section 5 -- Disclaimer of Warranties and Limitation of Liability.
285
+
286
+ a. UNLESS OTHERWISE SEPARATELY UNDERTAKEN BY THE LICENSOR, TO THE
287
+ EXTENT POSSIBLE, THE LICENSOR OFFERS THE LICENSED MATERIAL AS-IS
288
+ AND AS-AVAILABLE, AND MAKES NO REPRESENTATIONS OR WARRANTIES OF
289
+ ANY KIND CONCERNING THE LICENSED MATERIAL, WHETHER EXPRESS,
290
+ IMPLIED, STATUTORY, OR OTHER. THIS INCLUDES, WITHOUT LIMITATION,
291
+ WARRANTIES OF TITLE, MERCHANTABILITY, FITNESS FOR A PARTICULAR
292
+ PURPOSE, NON-INFRINGEMENT, ABSENCE OF LATENT OR OTHER DEFECTS,
293
+ ACCURACY, OR THE PRESENCE OR ABSENCE OF ERRORS, WHETHER OR NOT
294
+ KNOWN OR DISCOVERABLE. WHERE DISCLAIMERS OF WARRANTIES ARE NOT
295
+ ALLOWED IN FULL OR IN PART, THIS DISCLAIMER MAY NOT APPLY TO YOU.
296
+
297
+ b. TO THE EXTENT POSSIBLE, IN NO EVENT WILL THE LICENSOR BE LIABLE
298
+ TO YOU ON ANY LEGAL THEORY (INCLUDING, WITHOUT LIMITATION,
299
+ NEGLIGENCE) OR OTHERWISE FOR ANY DIRECT, SPECIAL, INDIRECT,
300
+ INCIDENTAL, CONSEQUENTIAL, PUNITIVE, EXEMPLARY, OR OTHER LOSSES,
301
+ COSTS, EXPENSES, OR DAMAGES ARISING OUT OF THIS PUBLIC LICENSE OR
302
+ USE OF THE LICENSED MATERIAL, EVEN IF THE LICENSOR HAS BEEN
303
+ ADVISED OF THE POSSIBILITY OF SUCH LOSSES, COSTS, EXPENSES, OR
304
+ DAMAGES. WHERE A LIMITATION OF LIABILITY IS NOT ALLOWED IN FULL OR
305
+ IN PART, THIS LIMITATION MAY NOT APPLY TO YOU.
306
+
307
+ c. The disclaimer of warranties and limitation of liability provided
308
+ above shall be interpreted in a manner that, to the extent
309
+ possible, most closely approximates an absolute disclaimer and
310
+ waiver of all liability.
311
+
312
+
313
+ Section 6 -- Term and Termination.
314
+
315
+ a. This Public License applies for the term of the Copyright and
316
+ Similar Rights licensed here. However, if You fail to comply with
317
+ this Public License, then Your rights under this Public License
318
+ terminate automatically.
319
+
320
+ b. Where Your right to use the Licensed Material has terminated under
321
+ Section 6(a), it reinstates:
322
+
323
+ 1. automatically as of the date the violation is cured, provided
324
+ it is cured within 30 days of Your discovery of the
325
+ violation; or
326
+
327
+ 2. upon express reinstatement by the Licensor.
328
+
329
+ For the avoidance of doubt, this Section 6(b) does not affect any
330
+ right the Licensor may have to seek remedies for Your violations
331
+ of this Public License.
332
+
333
+ c. For the avoidance of doubt, the Licensor may also offer the
334
+ Licensed Material under separate terms or conditions or stop
335
+ distributing the Licensed Material at any time; however, doing so
336
+ will not terminate this Public License.
337
+
338
+ d. Sections 1, 5, 6, 7, and 8 survive termination of this Public
339
+ License.
340
+
341
+
342
+ Section 7 -- Other Terms and Conditions.
343
+
344
+ a. The Licensor shall not be bound by any additional or different
345
+ terms or conditions communicated by You unless expressly agreed.
346
+
347
+ b. Any arrangements, understandings, or agreements regarding the
348
+ Licensed Material not stated herein are separate from and
349
+ independent of the terms and conditions of this Public License.
350
+
351
+
352
+ Section 8 -- Interpretation.
353
+
354
+ a. For the avoidance of doubt, this Public License does not, and
355
+ shall not be interpreted to, reduce, limit, restrict, or impose
356
+ conditions on any use of the Licensed Material that could lawfully
357
+ be made without permission under this Public License.
358
+
359
+ b. To the extent possible, if any provision of this Public License is
360
+ deemed unenforceable, it shall be automatically reformed to the
361
+ minimum extent necessary to make it enforceable. If the provision
362
+ cannot be reformed, it shall be severed from this Public License
363
+ without affecting the enforceability of the remaining terms and
364
+ conditions.
365
+
366
+ c. No term or condition of this Public License will be waived and no
367
+ failure to comply consented to unless expressly agreed to by the
368
+ Licensor.
369
+
370
+ d. Nothing in this Public License constitutes or may be interpreted
371
+ as a limitation upon, or waiver of, any privileges and immunities
372
+ that apply to the Licensor or You, including from the legal
373
+ processes of any jurisdiction or authority.
374
+
375
+
376
+ =======================================================================
377
+
378
+ Creative Commons is not a party to its public
379
+ licenses. Notwithstanding, Creative Commons may elect to apply one of
380
+ its public licenses to material it publishes and in those instances
381
+ will be considered the “Licensor.” The text of the Creative Commons
382
+ public licenses is dedicated to the public domain under the CC0 Public
383
+ Domain Dedication. Except for the limited purpose of indicating that
384
+ material is shared under a Creative Commons public license or as
385
+ otherwise permitted by the Creative Commons policies published at
386
+ creativecommons.org/policies, Creative Commons does not authorize the
387
+ use of the trademark "Creative Commons" or any other trademark or logo
388
+ of Creative Commons without its prior written consent including,
389
+ without limitation, in connection with any unauthorized modifications
390
+ to any of its public licenses or any other arrangements,
391
+ understandings, or agreements concerning use of licensed material. For
392
+ the avoidance of doubt, this paragraph does not form part of the
393
+ public licenses.
394
+
395
+ Creative Commons may be contacted at creativecommons.org.
README.md ADDED
@@ -0,0 +1,247 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-4.0
3
+ base_model:
4
+ - Ultralytics/YOLOv8
5
+ pipeline_tag: object-detection
6
+ library_name: ultralytics
7
+ github: https://github.com/rfonod/geo-trax
8
+ language:
9
+ - en
10
+ tags:
11
+ - yolov8
12
+ - object-detection
13
+ - aerial-imagery
14
+ - drone
15
+ - vehicle-detection
16
+ - birds-eye-view
17
+ - geo-trax
18
+ - trajectory
19
+ - urban-traffic
20
+ - tracking
21
+ - arxiv:2411.02136
22
+ datasets:
23
+ - rfonod/songdo-vision
24
+ - Voxel51/VisDrone2019-DET
25
+ - detection-datasets/coco
26
+ model-index:
27
+ - name: rfonod/geo-trax/geotrax_hbb_yolov8s_1920_v1
28
+ results:
29
+ - task:
30
+ type: object-detection
31
+ dataset:
32
+ type: rfonod/songdo-vision
33
+ name: Songdo Vision
34
+ split: test
35
+ metrics:
36
+ - type: precision # mAP@0.5 not available as a standard metric type on HF
37
+ value: 0.951
38
+ name: mAP@0.5
39
+ - type: precision
40
+ value: 0.711
41
+ name: mAP@0.5:0.95
42
+ - type: precision
43
+ value: 0.911
44
+ name: Precision
45
+ - type: recall
46
+ value: 0.935
47
+ name: Recall
48
+ ---
49
+
50
+ # Geo-trax: YOLOv8s Vehicle Detector for Drone BEV Imagery
51
+
52
+ [![GitHub](https://img.shields.io/badge/GitHub-geo--trax-blue?logo=github)](https://github.com/rfonod/geo-trax)
53
+ [![PyPI](https://img.shields.io/pypi/v/geo-trax)](https://pypi.org/project/geo-trax/)
54
+ [![Paper](https://img.shields.io/badge/Journal-10.1016%2Fj.trc.2025.105205-blue)](https://doi.org/10.1016/j.trc.2025.105205)
55
+ [![arXiv](https://img.shields.io/badge/arXiv-2411.02136-b31b1b)](https://arxiv.org/abs/2411.02136)
56
+ [![Zenodo](https://img.shields.io/badge/Zenodo-Software%20Archive-blue)](https://zenodo.org/doi/10.5281/zenodo.12119542)
57
+ [![YouTube](https://img.shields.io/badge/YouTube-Demo-red?logo=youtube&logoColor=red)](https://youtu.be/gOGivL9FFLk)
58
+ [![Website](https://img.shields.io/badge/REAL%20Lab-Geo--trax-informational)](https://www.real-lab.ch/geo-trax)
59
+ [![License](https://img.shields.io/badge/License-CC%20BY%204.0-lightgrey)](LICENSE)
60
+
61
+ This is the default detection model for **[Geo-trax](https://github.com/rfonod/geo-trax)**, a
62
+ comprehensive pipeline for extracting georeferenced vehicle trajectories from high-altitude drone
63
+ (bird's-eye view) video footage. The model detects vehicles in aerial imagery and underpins the
64
+ results reported in the associated [publication](https://doi.org/10.1016/j.trc.2025.105205).
65
+
66
+ ![Geo-trax Output Visualization](https://raw.githubusercontent.com/rfonod/geo-trax/v1.0.0/assets/geo-trax_visualization.webp)
67
+
68
+ 🎬 This accelerated animation previews some of the capabilities of Geo-trax. Watch the full
69
+ demonstration (~4 min) on [YouTube](https://youtu.be/gOGivL9FFLk).
70
+
71
+ ## Model Details
72
+
73
+ | Property | Value |
74
+ |---|---|
75
+ | Architecture | YOLOv8s (HBB, horizontal bounding boxes) |
76
+ | Input resolution | 1920 × 1920 px |
77
+ | Classes | 6 trained (4 primary + 2 auxiliary; see below) |
78
+ | Parameters | ~11 M |
79
+ | Framework | [Ultralytics](https://github.com/ultralytics/ultralytics) ≥ 8.4.64 |
80
+ | Trained on | 19,339 annotated aerial images (679,306 labeled instances); multi-stage, see [publication](https://doi.org/10.1016/j.trc.2025.105205) |
81
+ | Validated on | [Songdo Vision](https://doi.org/10.5281/zenodo.13828407) test set (1,084 images, 55,124 vehicle instances) |
82
+
83
+ ### Classes and Detection Performance
84
+
85
+ Metrics reported on the [Songdo Vision](https://doi.org/10.5281/zenodo.13828407) test split
86
+ (1,084 images, 55,124 labeled vehicle instances). The `Instances` column is the per-class
87
+ support in the test set. See Table 3 of the
88
+ [publication](https://doi.org/10.1016/j.trc.2025.105205) for full results.
89
+
90
+ | ID | Label | Notes | Instances | Precision | Recall | mAP@50 | mAP@50-95 |
91
+ |---|---|---|---:|---|---|---|---|
92
+ | 0 | Car | incl. vans | 49,508 | 0.979 | 0.981 | 0.992 | 0.835 |
93
+ | 1 | Bus | | 1,759 | 0.952 | 0.977 | 0.988 | 0.826 |
94
+ | 2 | Truck | | 3,052 | 0.887 | 0.916 | 0.935 | 0.722 |
95
+ | 3 | Motorcycle | | 805 | 0.827 | 0.866 | 0.888 | 0.463 |
96
+ | 4 | Pedestrian | not evaluated | n/a | n/a | n/a | n/a | n/a |
97
+ | 5 | Bicycle | not evaluated | n/a | n/a | n/a | n/a | n/a |
98
+ | **All** | | | **55,124** | **0.911** | **0.935** | **0.951** | **0.711** |
99
+
100
+ The model reaches **0.951 mAP@50** and **0.711 mAP@50-95** overall, with near-saturated accuracy
101
+ on cars and buses (mAP@50 ≥ 0.988). Trucks and especially motorcycles are harder: motorcycles are
102
+ small, sparse in the test set (805 instances), and the main driver of the lower mAP@50-95.
103
+
104
+ ### Evaluation Plots
105
+
106
+ Precision-recall curves and the normalized confusion matrix on the Songdo Vision test set:
107
+
108
+ <table>
109
+ <tr>
110
+ <td align="center"><img src="assets/PR_curve.png" alt="Precision-Recall Curve" width="420"><br><sub>Precision-Recall Curve</sub></td>
111
+ <td align="center"><img src="assets/confusion_matrix_normalized.png" alt="Normalized Confusion Matrix" width="420"><br><sub>Normalized Confusion Matrix</sub></td>
112
+ </tr>
113
+ </table>
114
+
115
+ > **Note on pedestrian and bicycle classes:** The model was trained on pedestrian and bicycle
116
+ > instances; however, these classes are **not evaluated and not recommended for use**. They were
117
+ > underrepresented in the training data, are not annotated in the Songdo Vision dataset (making
118
+ > reliable evaluation impossible), and achieve poor detection performance in practice.
119
+
120
+ ## How to Use
121
+
122
+ ### With Geo-trax (recommended)
123
+
124
+ This model is the default in Geo-trax and downloads automatically on first use:
125
+
126
+ ```bash
127
+ pip install geo-trax
128
+ geotrax extract video.mp4 # detect, track, and stabilize; auto-downloads the model
129
+ geotrax batch video.mp4 --no-geo # detect, track, and stabilize; skip georeferencing
130
+ geotrax batch video.mp4 # full pipeline including georeferencing (requires orthophotos)
131
+ ```
132
+
133
+ See the Geo-trax [GitHub README](https://github.com/rfonod/geo-trax) for the full pipeline,
134
+ configuration options, and georeferencing.
135
+
136
+ ### Direct Ultralytics inference
137
+
138
+ ```python
139
+ from ultralytics import YOLO
140
+ from huggingface_hub import hf_hub_download
141
+
142
+ weights = hf_hub_download(repo_id="rfonod/geo-trax", filename="geotrax_hbb_yolov8s_1920_v1.pt")
143
+ model = YOLO(weights)
144
+
145
+ results = model("drone_frame.jpg", imgsz=1920, conf=0.25, iou=0.45, classes=[0, 1, 2, 3])
146
+ results[0].show()
147
+ ```
148
+
149
+ > **Tip:** The model was trained and validated at **1920 px** input resolution. Downscaling to
150
+ > 1280 px is possible with a small accuracy trade-off; going below 960 px significantly degrades
151
+ > detection of small vehicles (motorcycles, distant cars). Pass `classes=[0, 1, 2, 3]` to
152
+ > restrict inference to the four evaluated classes and suppress unreliable predictions.
153
+
154
+ ## Training Data
155
+
156
+ Training followed a multi-stage strategy starting from **YOLOv8s weights pretrained on COCO**
157
+ as the initial foundation. Two successive stages were applied:
158
+
159
+ **Stage 1 (BASE):** The model was trained on a large, diverse collection drawn from eight
160
+ public aerial and drone datasets (CARPK, PUCPR+, CyCAR, UAVDT, HARPY, RAI4VD, UIT-ADrone,
161
+ and VisDrone) combined with the [Songdo Vision](https://doi.org/10.5281/zenodo.13828407)
162
+ dataset, totalling 19,339 training images with 679,306 annotations across 6 vehicle classes
163
+ (car, bus, truck, motorcycle, pedestrian, bicycle).
164
+
165
+ **Stage 2 (FINE):** The BASE-trained model was subsequently fine-tuned on a curated,
166
+ high-quality subset of 9,004 images with 321,368 annotations, emphasising accurate annotations
167
+ and higher-resolution images, again combined with
168
+ [Songdo Vision](https://doi.org/10.5281/zenodo.13828407), to yield the final weights released here.
169
+
170
+ **Training set composition** (annotations per class):
171
+
172
+ | Stage | Images | Annotations | Car | Bus | Truck | Motorcycle | Pedestrian | Bicycle |
173
+ |---|---:|---:|---:|---:|---:|---:|---:|---:|
174
+ | BASE | 19,339 | 679,306 | 561,666 | 15,587 | 28,830 | 44,512 | 24,239 | 4,472 |
175
+ | FINE | 9,004 | 321,368 | 266,745 | 8,047 | 14,305 | 30,925 | 1,260 | 86 |
176
+
177
+ [Songdo Vision](https://doi.org/10.5281/zenodo.13828407) comprises 5,419 annotated drone frames
178
+ (4,335 training / 1,084 test; 80/20 split) collected during a large-scale urban traffic
179
+ monitoring experiment in Songdo, South Korea. It covers four primary vehicle classes captured
180
+ at 140-150 m altitude by DJI Mavic 3 drones, contributing 217,311 training and 55,124 test
181
+ instances to the totals above.
182
+
183
+ **Training configuration:**
184
+
185
+ | Setting | Value |
186
+ |---|---|
187
+ | Initialization | YOLOv8s pretrained on COCO |
188
+ | Optimizer | SGD |
189
+ | Learning rate (initial / final factor) | 0.01 / 0.01 |
190
+ | Momentum | 0.937 |
191
+ | Weight decay | 0.0005 |
192
+ | Batch size | 8 |
193
+ | Early stopping | 50-epoch patience |
194
+ | Input resolution | 1920 × 1920 px (letterbox padding) |
195
+ | Mixed precision | AMP enabled |
196
+ | Augmentation | random scaling, translation, horizontal flip, mosaic, colour jitter, Gaussian/median blur, grayscale, CLAHE |
197
+
198
+ See the [publication](https://doi.org/10.1016/j.trc.2025.105205) for complete dataset statistics,
199
+ training details, and ablation results.
200
+
201
+ ## Intended Use and Limitations
202
+
203
+ - **GSD assumption:** The bundled Geo-trax config assumes a ground sampling distance (GSD) of
204
+ ~0.027 m/px (DJI Mavic 3, 4K, 140-150 m altitude). Adjust this value in the config for
205
+ different hardware or flight altitudes.
206
+ - **Supported classes:** Car, bus, truck, and motorcycle (class IDs 0-3). The model was also
207
+ trained on pedestrian and bicycle instances; however, these classes achieve poor detection
208
+ performance and are not recommended for use (see the class table above). Geo-trax filters to
209
+ the four primary classes by default; when using Ultralytics directly, pass
210
+ `classes=[0, 1, 2, 3]` to suppress unreliable predictions.
211
+
212
+ ## Citation
213
+
214
+ If you use this model, please cite the associated publication:
215
+
216
+ ```bibtex
217
+ @article{fonod2025advanced,
218
+ title = {Advanced computer vision for extracting georeferenced vehicle trajectories from drone imagery},
219
+ author = {Fonod, Robert and Cho, Haechan and Yeo, Hwasoo and Geroliminis, Nikolas},
220
+ journal = {Transportation Research Part C: Emerging Technologies},
221
+ volume = {178},
222
+ pages = {105205},
223
+ year = {2025},
224
+ doi = {10.1016/j.trc.2025.105205}
225
+ }
226
+ ```
227
+
228
+ If you additionally use the [Geo-trax software](https://github.com/rfonod/geo-trax), please
229
+ also cite the specific version you used via its Zenodo record. For example, for version 1.0.0:
230
+
231
+ ```bibtex
232
+ @software{fonod2026geo-trax,
233
+ author = {Fonod, Robert},
234
+ title = {Geo-trax: A Comprehensive Framework for Georeferenced Vehicle Trajectory Extraction from Drone Imagery},
235
+ url = {https://github.com/rfonod/geo-trax},
236
+ doi = {10.5281/zenodo.12119542},
237
+ version = {1.0.0},
238
+ year = {2026}
239
+ }
240
+ ```
241
+
242
+ ## License
243
+
244
+ This model is released under the
245
+ [Creative Commons Attribution 4.0 International (CC BY 4.0)](https://creativecommons.org/licenses/by/4.0/)
246
+ license; see the [LICENSE](LICENSE) file for the full terms. The Geo-trax codebase is distributed
247
+ separately under the [MIT License](https://github.com/rfonod/geo-trax/blob/main/LICENSE).
assets/PR_curve.png ADDED

Git LFS Details

  • SHA256: a97a1e0544d2135f0cff0f2d5d8d65f1ec797118d258dda0c5f53fe8dda4b38a
  • Pointer size: 131 Bytes
  • Size of remote file: 162 kB
assets/confusion_matrix_normalized.png ADDED

Git LFS Details

  • SHA256: 6b1cf0838eacbddf692b30e1e5c1800fc78c766e5b19069908ee03484f9bba60
  • Pointer size: 131 Bytes
  • Size of remote file: 142 kB
geotrax_hbb_yolov8s_1920_v1.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d462ae523b15f3679a83f74ab08aa79d2978f319ca2f0cf2892fbbb60df79da
3
+ size 22927726