CoolFace
Modelpublic

dronefreak/uavdt-yolov9t

sourceHugging Faceagpl-3.0updated 4d agoView on Hugging Face
0likes36downloads
Model Card

YOLOv9t Finetuned on UAVDT

Fine-tuned YOLOv9t object detector on the UAVDT benchmark dataset, trained and evaluated as part of DetectionBench -- a framework for reproducibly benchmarking modern object detectors with identical training recipes and evaluation metrics across multiple real-world datasets.

<!-- Demo banner: side-by-side video of this checkpoint's detections on two UAVDT test clips. Media lives under assets/ in this repo. The <video> renders on the Hugging Face model page (absolute resolve/ URL); on GitHub the nested <img> poster is shown instead. --> <p align="center"><video controls autoplay loop muted playsinline width="900" poster="https://huggingface.co/dronefreak/uavdt-yolov9t/resolve/main/assets/demobannerposter.jpg" src="https://huggingface.co/dronefreak/uavdt-yolov9t/resolve/main/assets/demobanner.mp4"><img src="https://huggingface.co/dronefreak/uavdt-yolov9t/resolve/main/assets/demobanner_poster.jpg" alt="YOLOv9t detections on two UAVDT test clips" width="900"></video></p>

<br>

<!-- ROW 1: Identity & Tech Stack --> <div style="display: flex; justify-content: center; align-items: center; gap: 8px; margin-bottom: 8px; flex-wrap: wrap;"> <img src="https://img.shields.io/badge/Task-ObjectDetection-blue?style=flat-square" alt="Task"> <img src="https://img.shields.io/badge/Framework-UltralyticsYOLO-0aa1a7?style=flat-square" alt="Framework"> <img src="https://img.shields.io/badge/Base_Model-YOLOv9t-purple?style=flat-square" alt="Base Model"> </div>

<!-- ROW 2: Performance Metrics --> <div style="display: flex; justify-content: center; align-items: center; gap: 8px; margin-bottom: 8px; flex-wrap: wrap;"> <img src="https://img.shields.io/badge/mAP@50-29.42%25-success?style=flat-square" alt="mAP@50"> <img src="https://img.shields.io/badge/mAP@50:95-17.03%25-orange?style=flat-square" alt="mAP@50:95"> <img src="https://img.shields.io/badge/Params-2.1M-lightgrey?style=flat-square" alt="Params"> </div>

<!-- ROW 3: Metadata --> <div style="display: flex; justify-content: center; align-items: center; gap: 8px; margin-bottom: 24px; flex-wrap: wrap;"> <img src="https://img.shields.io/badge/License-AGPL--3.0-lightgrey?style=flat-square" alt="License"> <a href="https://github.com/dronefreak/DetectionBench"><img src="https://img.shields.io/badge/Source-DetectionBench-black?style=flat-square" alt="Source"></a> </div>


Performance

MetricScore (%)
mAP@5029.42
mAP@50-9517.03
Precision35.75
Recall36.47
F1 Score36.1
Parameters2.1M
FLOPs8.5B (at 640 px)

Evaluation Protocol

Metrics reported in this model card are computed on the UAVDT test split, using DetectionBench's standard evaluation pipeline (detectionbench-evaluate).


UAVDT Model Zoo

Every model DetectionBench has trained and evaluated on UAVDT so far, for full transparency -- see DetectionBench for the smaller, curated comparison set used on the project README.

ModelmAP@50mAP@50-95PrecisionRecall
YOLOv26m33.4319.5638.1439.84
RF-DETR Medium33.2820.5473.0370.03
YOLOv26s32.9819.6143.8640.38
RF-DETR Nano32.7820.3173.666.98
RF-DETR Small32.6220.2173.8371.63
YOLOv9s31.8218.7139.8338.12
YOLOv8m31.4218.840.2737.79
YOLOv11m30.4717.7137.737.01
YOLOv10m30.1217.3340.1335.68
YOLOv9m29.4316.9735.9235.7
YOLOv9t29.4217.0335.7536.47
YOLOv11s29.117.1634.3237.31
YOLOv26n28.8816.7933.1435.66
YOLOv10s28.8516.4836.5333.16
YOLOv11n28.5616.338.0432.26
YOLOv8n27.815.3435.4233.61
YOLOv10n27.1715.1633.331.21
YOLOv8s27.1215.3334.6531.87

Per-Class Performance

ClassmAP@50mAP@50-95
car69.5638.11
truck3.582.34
bus15.1110.63

Evaluation Visualizations

Precision-Recall Curve

[image]

F1 Curve

[image]

Confusion Matrix

[image]

Normalized Confusion Matrix

[image]


Dataset

This model was trained on UAVDT. For the full dataset description, provenance, license, and citation, see the dataset card:

https://huggingface.co/datasets/dronefreak/UAVDT

Classes

  • —car
  • —truck
  • —bus ---

Usage

Install Dependencies

bash
pip install ultralytics huggingface_hub

Load Model from Hugging Face

python
from huggingface_hub import hf_hub_download
from ultralytics import YOLO

weights = hf_hub_download(
    repo_id="dronefreak/uavdt-yolov9t",
    filename="best.pt"
)

model = YOLO(weights)

Run Inference

python
results = model.predict(
    source="image.jpg",
    conf=0.25
)

results[0].show()

Training Configuration

SettingValue
DatasetUAVDT
FrameworkUltralytics YOLO
Training ToolkitDetectionBench
Epochs (configured max)30
Epochs (actually trained)19
Early Stopping Patience8
Batch Sizeauto (Ultralytics AutoBatch)
Image Size1024
OptimizerAdamW
Initial Learning Rate0.0005
Seed0

Repository Contents

text
best.pt
results.csv
args.yaml
BoxPR_curve.png
BoxF1_curve.png
BoxP_curve.png
BoxR_curve.png
confusion_matrix.png
confusion_matrix_normalized.png
val_batch0_pred.jpg
uavdt_yolov9t_showcase.jpg
assets/demo_banner.mp4
assets/demo_banner_poster.jpg
README.md

Related Resources


Training Framework

This model was trained using DetectionBench, an open-source framework for benchmarking object detectors across multiple real-world datasets with a common pipeline.

Features include:

  • —A dataset-adapter registry for converting real-world datasets into a canonical format
  • —Identical training/evaluation recipes across model families (Ultralytics YOLO/RT-DETR, RF-DETR)
  • —Hardware profiling (latency, FPS, VRAM, parameters, FLOPs)
  • —One-command reproducibility via versioned Hydra configs

If you find this model useful, please consider starring the repository.


Known Limitations

  • —Severe class imbalance: car (94.6%) dominates the annotated boxes, while truck (3.1%) and bus (2.3%) are rare -- per-class accuracy on the minority classes is measured on comparatively few examples, and every model here scores far lower on them than on car.
  • —Very small objects: the median box covers only 0.14% of the image area (mean 0.26%), so this is a hard small-object regime and absolute mAP values are low for every architecture; the numbers are best read as a relative comparison between models, not as a production-quality detector.
  • —Video-derived, highly correlated frames: the ~40.7k labelled images come from 50 video sequences, so consecutive frames are near-duplicates. UAVDT's 50 tracking-only sequences have no detection labels and are excluded. The validation split is carved out of the training sequences by sequence (not by frame) to avoid leakage, but effective diversity is far lower than the image count suggests.
  • —Different density per split: instances per image are 15.7 (train), 28.0 (valid) and 22.7 (test), because the splits contain different sequences -- validation metrics are not directly predictive of test metrics.
  • —Research-use-only data: UAVDT is distributed "for research purpose only" with no redistribution grant, so the dataset is not mirrored here -- obtain it from the official source (see the Dataset section above) and check its terms before any use beyond research. ---

Citation

If you use this model in your research, please consider citing:

  1. 1.The UAVDT dataset (see below)
  2. 2.The original YOLOv9t architecture (see below)
  3. 3.The other model architectures shown in the Model Zoo/External Comparison tables above, if you reference their results
  4. 4.DetectionBench, the training/evaluation framework used to produce this checkpoint
@InProceedings{du2018unmanned,
  title={The Unmanned Aerial Vehicle Benchmark: Object Detection and Tracking},
  author={Du, Dawei and Qi, Yuankai and Yu, Hongyang and Yang, Yifan and Duan, Kaiwen and Li, Guorong and Zhang, Weigang and Huang, Qingming and Tian, Qi},
  booktitle={Proceedings of the European Conference on Computer Vision (ECCV)},
  year={2018}
}
bibtex
@article{wang2024yolov9,
  title={YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information},
  author={Wang, Chien-Yao and Yeh, I-Hau and Liao, Hong-Yuan Mark},
  journal={arXiv preprint arXiv:2402.13616},
  year={2024}
}

Other architectures compared against on UAVDT in this model card:

RF-DETR

bibtex
@inproceedings{robinson2026rfdetr,
  title     = {RF-DETR: Real-Time Detection Transformer},
  author    = {Robinson, Isaac and Robicheaux, Peter and Popov, Matvei and Ramanan, Deva and Peri, Neehar},
  booktitle = {International Conference on Learning Representations (ICLR)},
  year      = {2026},
  url       = {https://arxiv.org/abs/2511.09554}
}

@article{oquab2023dinov2,
  title={DINOv2: Learning Robust Visual Features without Supervision},
  author={Oquab, Maxime and Darcet, Timoth{\'e}e and Moutakanni, Theo and Vo, Huy and Szafraniec, Marc and Khalidov, Vasil and Fernandez, Pierre and Haziza, Daniel and Massa, Francisco and El-Nouby, Alaaeldin and others},
  journal={arXiv preprint arXiv:2304.07193},
  year={2023}
}

YOLOv10

bibtex
@article{wang2024yolov10,
  title={YOLOv10: Real-Time End-to-End Object Detection},
  author={Wang, Ao and Chen, Hui and Liu, Lihao and Chen, Kai and Lin, Zijia and Han, Jungong and Ding, Guiguang},
  journal={arXiv preprint arXiv:2405.14458},
  year={2024}
}

YOLOv11

bibtex
No official YOLO11 research paper has been published by Ultralytics; the most commonly cited independent architectural analysis is used instead:

@article{khanam2024yolov11,
  title={YOLOv11: An Overview of the Key Architectural Enhancements},
  author={Khanam, Rahima and Hussain, Muhammad},
  journal={arXiv preprint arXiv:2410.17725},
  year={2024}
}

YOLOv26

bibtex
@article{jocher2026yolo26,
  title={Ultralytics YOLO26: Unified Real-Time End-to-End Vision Models},
  author={Jocher, Glenn and Qiu, Jing and Liu, Mengyu and Lyu, Shuai and Akyon, Fatih Cagatay and Kalfaoglu, Muhammet Esat},
  journal={arXiv preprint arXiv:2606.03748},
  year={2026}
}

YOLOv8

bibtex
No official YOLOv8 research paper has been published by Ultralytics; this is their own recommended software citation instead:

@software{jocher2023yolov8,
  author = {Glenn Jocher and Ayush Chaurasia and Jing Qiu},
  title = {Ultralytics YOLOv8},
  version = {8.0.0},
  year = {2023},
  url = {https://github.com/ultralytics/ultralytics},
  license = {AGPL-3.0}
}
bibtex
@software{Saksena_DetectionBench_2026,
  author = {Saksena, Saumya Kumaar},
  title = {DetectionBench: Reproducible Benchmarks for Modern Object Detectors on Real-World Datasets},
  url = {https://github.com/dronefreak/DetectionBench},
  year = {2026}
}