rfonod commited on
Commit
c169663
·
verified ·
1 Parent(s): 3285aae

Fix webp animation link to point to main branch; bump citation example to v1.3.0

Browse files
Files changed (1) hide show
  1. README.md +5 -26
README.md CHANGED
@@ -2,15 +2,12 @@
2
  license: cc-by-4.0
3
  base_model:
4
  - Ultralytics/YOLOv8
5
- base_model_relation: finetune
6
  pipeline_tag: object-detection
7
  library_name: ultralytics
8
- num_parameters: 11137922
9
  github: https://github.com/rfonod/geo-trax
10
  language:
11
  - en
12
  tags:
13
- - ultralytics
14
  - yolov8
15
  - object-detection
16
  - aerial-imagery
@@ -68,7 +65,7 @@ comprehensive pipeline for extracting georeferenced vehicle trajectories from hi
68
  (bird's-eye view) video footage. The model detects vehicles in aerial imagery and underpins the
69
  results reported in the associated [publication](https://doi.org/10.1016/j.trc.2025.105205).
70
 
71
- ![Geo-trax Output Visualization](https://raw.githubusercontent.com/rfonod/geo-trax/v1.0.0/assets/geo-trax_visualization.webp)
72
 
73
  🎬 This accelerated animation previews some of the capabilities of Geo-trax. Watch the full
74
  demonstration (~4 min) on [YouTube](https://youtu.be/gOGivL9FFLk).
@@ -80,7 +77,7 @@ demonstration (~4 min) on [YouTube](https://youtu.be/gOGivL9FFLk).
80
  | Architecture | YOLOv8s (HBB, horizontal bounding boxes) |
81
  | Input resolution | 1920 × 1920 px |
82
  | Classes | 6 trained (4 primary + 2 auxiliary; see below) |
83
- | Parameters | 11.1 M (11,137,922) |
84
  | Framework | [Ultralytics](https://github.com/ultralytics/ultralytics) ≥ 8.4.64 |
85
  | Trained on | 19,339 annotated aerial images (679,306 labeled instances); multi-stage, see [publication](https://doi.org/10.1016/j.trc.2025.105205) |
86
  | Validated on | [Songdo Vision](https://doi.org/10.5281/zenodo.13828407) test set (1,084 images, 55,124 vehicle instances) |
@@ -140,7 +137,7 @@ geotrax batch video.mp4 # full pipeline including georeferencing (re
140
  See the Geo-trax [GitHub README](https://github.com/rfonod/geo-trax) for the full pipeline,
141
  configuration options, and georeferencing.
142
 
143
- ### Direct Ultralytics inference (PyTorch)
144
 
145
  ```python
146
  from ultralytics import YOLO
@@ -158,24 +155,6 @@ results[0].show()
158
  > detection of small vehicles (motorcycles, distant cars). Pass `classes=[0, 1, 2, 3]` to
159
  > restrict inference to the four evaluated classes and suppress unreliable predictions.
160
 
161
- ### ONNX inference
162
-
163
- An ONNX export (opset 12, static 1920 × 1920 input) is available for deployment without a PyTorch dependency:
164
-
165
- ```python
166
- import numpy as np
167
- import onnxruntime as ort
168
- from huggingface_hub import hf_hub_download
169
-
170
- onnx_path = hf_hub_download(repo_id="rfonod/geo-trax", filename="geotrax_hbb_yolov8s_1920_v1.onnx")
171
- session = ort.InferenceSession(onnx_path, providers=["CPUExecutionProvider"])
172
-
173
- # Prepare input: BGR image resized/padded to 1920×1920, normalized to [0, 1]
174
- img = np.random.rand(1, 3, 1920, 1920).astype(np.float32) # replace with real image
175
- outputs = session.run(None, {"images": img})
176
- # outputs[0] shape: (1, 10, 75600) — 10 = 4 bbox coords + 6 class scores
177
- ```
178
-
179
  ## Training Data
180
 
181
  Training followed a multi-stage strategy starting from **YOLOv8s weights pretrained on COCO**
@@ -270,7 +249,7 @@ If you use this model, please cite the associated publication:
270
  ```
271
 
272
  If you additionally use the [Geo-trax software](https://github.com/rfonod/geo-trax), please
273
- also cite the specific version you used via its Zenodo record. For example, for version 1.0.0:
274
 
275
  ```bibtex
276
  @software{fonod2026geo-trax,
@@ -278,7 +257,7 @@ also cite the specific version you used via its Zenodo record. For example, for
278
  title = {Geo-trax: A Comprehensive Framework for Georeferenced Vehicle Trajectory Extraction from Drone Imagery},
279
  url = {https://github.com/rfonod/geo-trax},
280
  doi = {10.5281/zenodo.12119542},
281
- version = {1.0.0},
282
  year = {2026}
283
  }
284
  ```
 
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
 
65
  (bird's-eye view) video footage. The model detects vehicles in aerial imagery and underpins the
66
  results reported in the associated [publication](https://doi.org/10.1016/j.trc.2025.105205).
67
 
68
+ ![Geo-trax Output Visualization](https://raw.githubusercontent.com/rfonod/geo-trax/main/assets/geo-trax_visualization.webp)
69
 
70
  🎬 This accelerated animation previews some of the capabilities of Geo-trax. Watch the full
71
  demonstration (~4 min) on [YouTube](https://youtu.be/gOGivL9FFLk).
 
77
  | Architecture | YOLOv8s (HBB, horizontal bounding boxes) |
78
  | Input resolution | 1920 × 1920 px |
79
  | Classes | 6 trained (4 primary + 2 auxiliary; see below) |
80
+ | Parameters | ~11 M |
81
  | Framework | [Ultralytics](https://github.com/ultralytics/ultralytics) ≥ 8.4.64 |
82
  | Trained on | 19,339 annotated aerial images (679,306 labeled instances); multi-stage, see [publication](https://doi.org/10.1016/j.trc.2025.105205) |
83
  | Validated on | [Songdo Vision](https://doi.org/10.5281/zenodo.13828407) test set (1,084 images, 55,124 vehicle instances) |
 
137
  See the Geo-trax [GitHub README](https://github.com/rfonod/geo-trax) for the full pipeline,
138
  configuration options, and georeferencing.
139
 
140
+ ### Direct Ultralytics inference
141
 
142
  ```python
143
  from ultralytics import YOLO
 
155
  > detection of small vehicles (motorcycles, distant cars). Pass `classes=[0, 1, 2, 3]` to
156
  > restrict inference to the four evaluated classes and suppress unreliable predictions.
157
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
158
  ## Training Data
159
 
160
  Training followed a multi-stage strategy starting from **YOLOv8s weights pretrained on COCO**
 
249
  ```
250
 
251
  If you additionally use the [Geo-trax software](https://github.com/rfonod/geo-trax), please
252
+ also cite the specific version you used via its Zenodo record. For example, for version 1.3.0:
253
 
254
  ```bibtex
255
  @software{fonod2026geo-trax,
 
257
  title = {Geo-trax: A Comprehensive Framework for Georeferenced Vehicle Trajectory Extraction from Drone Imagery},
258
  url = {https://github.com/rfonod/geo-trax},
259
  doi = {10.5281/zenodo.12119542},
260
+ version = {1.3.0},
261
  year = {2026}
262
  }
263
  ```
Free AI Image Generator No sign-up. Instant results. Open Now