YOLO26m Finetuned on LISA Traffic Lights

Fine-tuned YOLO26m object detector on the LISA Traffic Lights 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.

LISA Traffic Lights Detection Demo

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/lisa-yolo26m",
    filename="best.pt"
)

model = YOLO(weights)

Run Inference

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

results[0].show()

Performance

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

Metric	Score (%)
mAP@50	29.08
mAP@50-95	13.68
Precision	43.79
Recall	29.71
F1 Score	35.4
Parameters	21.9M
FLOPs	75.4B (at 640 px)

LISA Traffic Lights Model Zoo

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

Model	mAP@50	mAP@50-95	Precision	Recall
RF-DETR Medium	33.01	14.12	69.37	58.28
RF-DETR Small	32.7	15.11	69.7	53.9
YOLO26m	29.08	13.68	43.79	29.71
YOLO26x	28.92	14.09	43.59	26.96
RF-DETR Nano	27.47	12.16	74.96	52.57
YOLO26l	27.28	13.54	41.42	27.99
YOLO26s	26.91	12.82	42.47	26.43
YOLO11x	26.4	13.09	53.98	24.38
YOLOv8m	25.07	11.91	38.16	25.15
YOLO26n	23.74	10.57	37.23	25.7

Per-Class Performance

Class	mAP@50	mAP@50-95
go	62.45	32.5
goForward	0.0	0.0
goLeft	7.26	2.84
stop	68.19	29.34
stopLeft	14.58	6.74
warning	49.74	23.8
warningLeft	1.34	0.55

Dataset

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

https://huggingface.co/datasets/dronefreak/LISA-Traffic-Lights

Classes

go
goForward
goLeft
stop
stopLeft
warning
warningLeft

Training Configuration

Setting	Value
Dataset	LISA Traffic Lights
Framework	Ultralytics YOLO
Training Toolkit	DetectionBench
Epochs (configured max)	500
Epochs (actually trained)	129
Early Stopping Patience	100
Batch Size	32
Image Size	640
Optimizer	auto
Initial Learning Rate	0.001
Seed	0

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

Related Resources

LISA Traffic Lights dataset card on Hugging Face
DetectionBench -- reproducible benchmarks for modern object detectors on real-world datasets

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

Rare/underrepresented arrow classes (goForward, goLeft, stopLeft, warningLeft) have far fewer training examples than the base go/stop/warning classes and correspondingly lower detection accuracy across every model in this zoo.
Sequential dashcam video frames mean visually similar consecutive frames can appear within the same split; performance on genuinely novel scenes may differ from the reported test-split numbers.
Trained and evaluated only on San Diego daytime/nighttime driving sequences (Pacific Beach, La Jolla); generalization to different traffic-light hardware, road layouts, or camera setups is untested.
Small, distant traffic lights are harder to detect reliably, consistent with general small-object detection challenges.

Citation

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

@article{jensen2016vision,
  title={Vision for looking at traffic lights: Issues, survey, and perspectives},
  author={Jensen, Morten Born{\o} and Philipsen, Mark Philip and M{\o}gelmose, Andreas and Moeslund, Thomas Baltzer and Trivedi, Mohan Manubhai},
  journal={IEEE Transactions on Intelligent Transportation Systems},
  volume={17},
  number={7},
  pages={1800--1815},
  year={2016},
  doi={10.1109/TITS.2015.2509509},
  publisher={IEEE}
}

@inproceedings{philipsen2015traffic,
  title={Traffic light detection: A learning algorithm and evaluations on challenging dataset},
  author={Philipsen, Mark Philip and Jensen, Morten Born{\o} and M{\o}gelmose, Andreas and Moeslund, Thomas B and Trivedi, Mohan M},
  booktitle={Intelligent Transportation Systems (ITSC), 2015 IEEE 18th International Conference on},
  pages={2341--2345},
  year={2015},
  organization={IEEE}
}

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