Ir para o conteúdo
ModelListadosExterno
Baixar

Citysample Vehicle Keypoints 24pt

A YOLO-pose model trained entirely on synthetic data - the City Sample 24-point vehicle-keypoint dataset rendered in Unreal Engine 5. It predicts a 24-point anatomical keypoint schema (wheels, head/tail lights,...

TipoOtherModelo baseUltralytics/YOLO26VersãoRevision 9aeced5Bnota

Fonte do modelo

Tipo
Other
Tamanho do arquivo
8,79 MB
Modelo base
Ultralytics/YOLO26
Versão
vRevision 9aeced5
Atualizado
10 de out. de 2026
Downloads
22
Fonte
Hugging Face

Descrição da fonte

City Sample Vehicle Keypoints - 24-point (synthetic-only)

A YOLO-pose model trained entirely on synthetic data - the City Sample 24-point vehicle-keypoint dataset rendered in Unreal Engine 5. It predicts a 24-point anatomical keypoint schema (wheels, head/tail lights, exhaust, roof corners, center, mirrors, bumper and window corners) plus a bounding box per vehicle.

Generated by kiselyovd/ue5-vehicle-synth.

What this model is for

This is a research / proof-of-concept model that demonstrates the synthetic dataset is clean and learnable: a model trained only on it localises vehicles and their keypoints well on held-out synthetic frames. For real-world 14-point vehicle keypoints, see the production model kiselyovd/vehicle-keypoints.

The training labels are derived geometrically from each vehicle's own mesh data (skeletal wheel bones, vertex-cloud roofline, light/mirror material sections), so keypoints are exact per body shape - a van's high tail lights, a pickup's cab-only roof - not a scaled sedan template.

In-domain results (held-out synthetic val)

MetricBoxPose
mAP@500.8400.602
mAP@50-950.6290.390

(Pose mAP is understated: ultralytics uses default OKS sigmas, which are tuned for 17-point human pose, not this 24-point vehicle schema.) Trained from yolo26n-pose on 2,259 synthetic frames (the mesh-derived 2,510-frame capture, 90/10 split), 100 epochs, imgsz 480.

Usage

from huggingface_hub import hf_hub_download
from ultralytics import YOLO

w = hf_hub_download("kiselyovd/citysample-vehicle-keypoints-24pt", "best.pt")
model = YOLO(w)
results = model.predict("your_street_scene.jpg")
# results[0].keypoints.xy -> (N, 24, 2) keypoints per detected vehicle

Honest caveats

  • Synthetic domain. Trained only on rendered frames; the sim-to-real gap is real and measured: on the CarFusion test set this model reaches [email protected] ~0.14 (first 14 points) - it learns vehicle topology but localisation on real photos is unreliable. A label-precision control (re-deriving all keypoints exactly from mesh geometry) left that number unchanged, so the gap is appearance and fleet coverage, not label quality.
  • Fleet coverage. City Sample has on the order of a dozen vehicle models; errors on real photos concentrate on body shapes far from that fleet. See the sim-to-real study for the full analysis.

License

MIT for the weights. Rendered training frames come from Epic's City Sample under the UE EULA (non-interactive renders are distributable; no Epic assets are shipped).

Fontes

1 fonte
Hugging FacePrimárioVersão 9aeced57eb48f361d9074593fb13b334323ee046
huggingface.co/kiselyovd/citysample-vehicle-keypoints-24pt/tree/9aeced57eb48f361d9074593fb13b334323ee046

Verificado 10 de out.

Artefatos de modelo

1 artefato

best.pt

pt · 8,79 MB · SHA-256 16d9793c283a…8379 · Hugging Face

Download file

Trechos de fonte

3 trechos

--- license: mit library_name: ultralytics pipeline_tag: keypoint-detection tags: - ultralytics - yolo - yolo-pose - keypoint-detection - vehicle - synthetic - unreal-engine-5 - city-sample - sim-to-real datasets: - kiselyovd/citysample-vehicle-keypoints-24pt base_model: Ultralytics/YOLO26 --- # City Sample Vehicle Keypoints - 24-point (synthetic-only) A YOLO-pose model trained **entirely on synthetic data** - the [City Sample 24-point vehicle-keypoint dataset](https://huggingface.co/datasets/kiselyovd/citysample-vehicle-keypoints-24pt) rendered in Unreal Engine 5. It predicts a 24-point anatomical keypoint schema (wheels, head/tail lights, exhaust, roof corners, center, mirrors, bumper and window corners) plus a bounding box per vehicle. Generated by [kiselyovd/ue5-vehicle-synth](https://github.com/kiselyovd/ue5-vehicle-synth). ## What this model is for This is a **research / proof-of-concept** model that demonstrates the synthetic dataset is clean and learnable: a model trained only on it localises vehicles and their keypoints well on held-out synthetic frames. For real-world 14-point vehicle keypoints, see the production model [kiselyovd/vehicle-keypoints](https://huggingface.co/kiselyovd/vehicle-keypoints). The training labels are derived geometrically from each vehicle's own mesh data (skeletal wheel bones, vertex-cloud roofline, light/mirror material sections), so keypoints are exact per body shape - a van's high tail lights, a pickup's cab-only roof - not a scaled sedan template. ## In-domain results (held-out synthetic val) | Metric | Box | Pose | |---|---|---| | mAP@50 | **0.840** | 0.602 | | mAP@50-95 | 0.629 | 0.390 | (Pose mAP is understated: ultralytics uses default OKS sigmas, which are tuned for 17-point human pose, not this 24-point vehicle schema.) Trained from `yolo26n-pose` on 2,259 synthetic frames (the mesh-derived 2,510-frame capture, 90/10 split), 100 epochs, imgsz 480. ## Usage ```python from huggingface_hub import hf_hub_download from ultralytics import YOLO w = hf_hub_download("kiselyovd/citysample-vehicle-keypoints-24pt", "best.pt") model = YOLO(w) results = model.predict("your_street_scene.jpg") # results[0].keypoints.xy -> (N, 24, 2) keypoints per detected vehicle ``` ## Honest caveats - **Synthetic domain.** Trained only on rendered frames; the sim-to-real gap is re...

Source context: 22 downloads · 0 likes · Pipeline keypoint-detection · Library ultralytics · Repo kiselyovd/citysample-vehicle-keypoints-24pt

Source context: 74 downloads · 0 likes · Pipeline keypoint-detection · Library ultralytics · Repo kiselyovd/citysample-vehicle-keypoints-24pt