dronefreak / bdd100k-yolov8m

YOLOv8m Finetuned on BDD100K

Fine-tuned YOLOv8m object detector on the BDD100K 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.

---

Usage

Install Dependencies

pip install ultralytics huggingface_hub

Load Model from Hugging Face

from huggingface_hub import hf_hub_download
from ultralytics import YOLO

weights = hf_hub_download(
    repo_id="dronefreak/bdd100k-yolov8m",
    filename="best.pt"
)

model = YOLO(weights)

Run Inference

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

results[0].show()

---

Performance

Evaluated on the BDD100K test split, using DetectionBench's standard evaluation pipeline ( detectionbench-evaluate ).

---

BDD100K Model Zoo

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

---

Per-Class Performance

---

Dataset

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

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

Classes

• person
• rider
• car
• truck
• bus
• train
• motor
• bike
• traffic light
• traffic sign

---

Training Configuration

---

Repository Contents

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
bdd100k_yolov8m_showcase.jpg
README.md

---

Related Resources

• BDD100K dataset card on Hugging Face
• DetectionBench -- reproducible benchmarks for modern object detectors on real-world datasets
• BDD100K project website (official data source)

---

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

• Not comparable to the official BDD100K test-server leaderboard: the official test split has no released labels, so the test split here is BDD100K's official validation set (10,000 images) and a seeded 15% slice of the official train set is held out for validation.
• Severe class imbalance: car (55.4%), traffic sign (18.6%) and traffic light (14.5%) dominate the boxes, while rider (0.4%), motor (0.2%) and especially train (about 150 boxes in the whole dataset) are rare -- per-class accuracy on those classes is measured on very few examples and is close to noise for train .
• Small objects: the median box covers only 0.09% of the 1280x720 frame, and traffic lights and signs are the smallest and hardest classes (medians of roughly 16 px and 21 px at native resolution), so scores on them depend heavily on input resolution.
• Detection labels only: BDD100K's lane-marking and drivable-area annotations are dropped, so these models cover the 2D object detection task only.
• Conditions are not broken down: the images span weather, time-of-day and scene conditions, but scores here are aggregated over all of them, and generalization outside the US road scenes BDD100K covers is untested.
• Non-commercial redistribution only: BDD100K's own data license (reproduced in full on the dataset card) permits redistribution for educational, research and not-for-profit purposes with notice -- commercial use needs separate permission from UC Berkeley's Office of Technology Licensing, not a blanket no-redistribution grant.
• Label vintage is 2018, not the paper's 2020 publication date: the detection labels mirrored here are BDD100K's original 2018 annotation release (matching the data license's own copyright year), carried through a two-hop Kaggle mirror -- see the dataset card for why a separate, later-dated label release couldn't be located or verified.

---

Citation

If you use this model in your research, please consider citing the dataset and the model architecture:

@inproceedings{yu2020bdd100k,
  title={BDD100K: A Diverse Driving Dataset for Heterogeneous Multitask Learning},
  author={Yu, Fisher and Chen, Haofeng and Wang, Xin and Xian, Wenqi and Chen, Yingying and Liu, Fangchen and Madhavan, Vashisht and Darrell, Trevor},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={2636--2645},
  year={2020}
}

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}
}