dronefreak/uavid-yolo26n-seg

YOLO26n-sem Finetuned on UAVid

Fine-tuned YOLO26n semantic segmentation model for aerial UAV imagery using the UAVid benchmark dataset.

This model is part of the UAVid Semantic Segmentation Model Zoo, a collection of CABiNet and YOLO26 models trained and evaluated under a common pipeline for aerial semantic segmentation.

---

Performance

Metric	Score
mIoU	63.58
Pixel Accuracy	76.13
Parameters	TODO
FLOPs	TODO

---

UAVid Model Zoo

Rank	Model	mIoU (%)	Pixel Acc (%)	Params	FLOPs
1	YOLO26m-sem	67.66	79.75	TODO	TODO
2	YOLO26l-sem	67.2	78.63	TODO	TODO
3	YOLO26s-sem	66.88	79.0	TODO	TODO
4	YOLO26n-sem	63.58	76.13	TODO	TODO
5	CABiNet (MobileNetV3-Large)	TBD	TBD	TODO	TODO
6	CABiNet (MobileNetV3-Small)	TBD	TBD	TODO	TODO
7	YOLO26x-sem	TBD	TBD	TODO	TODO

---

Per-Class IoU (%)

Class	YOLO26m-sem	YOLO26l-sem	YOLO26s-sem	YOLO26n-sem	CABiNet (MobileNetV3-Large)	CABiNet (MobileNetV3-Small)	YOLO26x-sem
Clutter	65.3	66.0	63.3	60.9	TBD	TBD	TBD
Building	91.3	91.5	91.0	88.7	TBD	TBD	TBD
Road	79.2	80.4	77.5	76.8	TBD	TBD	TBD
Static Car	63.0	60.5	62.5	57.1	TBD	TBD	TBD
Tree	76.5	76.2	75.3	73.1	TBD	TBD	TBD
Vegetation	68.4	67.3	66.9	63.2	TBD	TBD	TBD
Human	31.3	29.7	30.5	25.2	TBD	TBD	TBD
Moving Car	66.3	66.0	66.5	63.3	TBD	TBD	TBD

---

Evaluation Visualizations

Per-Class IoU Bar Chart

Confusion Matrix

Loss Curves

---

Dataset

UAVid is a high-resolution UAV semantic segmentation benchmark of urban street scenes, captured from oblique aerial viewpoints along street-side flight paths.

Classes

• Clutter
• Building
• Road
• Static Car
• Tree
• Vegetation
• Human
• Moving Car

---

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/yolo26n-sem-uavid",
    filename="best.pt"
)

model = YOLO(weights)

Run Inference

results = model.predict(source="image.png", task="semantic", imgsz=1024)
mask = results[0].semantic_mask.cpu().numpy().data  # (H, W) class-ID map

---

Training Configuration

Setting	Value
Epochs	500
Image size	1024
Batch size	8
Dataset	UAVid (converted images/+masks/ format)
Framework	Ultralytics YOLO
cls_pw (class weighting)	0.5

---

Official Resources

• UAVid Semantic Segmentation Model Zoo: https://huggingface.co/collections/dronefreak/uavid-semantic-segmentation-model-zoo
• CABiNet repository: https://github.com/dronefreak/CABiNet
• CABiNet Paper: https://arxiv.org/abs/2011.00993v2
• Official UAVid Website: https://uavid.nl/
• UAVid Dataset Archive: https://doi.org/10.17026/dans-x9f-w9sa
• UAVid Paper: https://arxiv.org/abs/1810.10438
• UAVid Published Journal: https://doi.org/10.1016/j.isprsjprs.2020.05.009
• Ultralytics YOLO: https://github.com/ultralytics/ultralytics
• Ultralytics YOLO26 Paper: https://arxiv.org/abs/2606.03748

---

Training Framework

Trained with the CABiNet repository, which pairs its own real-time segmentation trainer with a parallel Ultralytics YOLO26-sem pipeline — shared UAVid dataset tooling, training/eval, and mIoU benchmarking across both. Star the repo if you find these models useful!

---

Known Limitations

Performance may degrade in:

• Very small or thin objects (e.g. pedestrians, moving cars at altitude)
• Heavy occlusion under tree canopy
• Motion blur on moving vehicles
• Mixed/very high input resolutions (UAVid source images are 3840x2160 / 4096x2160; both pipelines evaluate at reduced imgsz)

---

Citation

Please cite the following:

@article{LYU2020108,
    author = "Ye Lyu and George Vosselman and Gui-Song Xia and Alper Yilmaz and Michael Ying Yang",
    title = "UAVid: A semantic segmentation dataset for UAV imagery",
    journal = "ISPRS Journal of Photogrammetry and Remote Sensing",
    volume = "165",
    pages = "108 - 119",
    year = "2020",
    issn = "0924-2716",
    doi = "https://doi.org/10.1016/j.isprsjprs.2020.05.009",
    url = "http://www.sciencedirect.com/science/article/pii/S0924271620301295",
}

@INPROCEEDINGS{9560977,
  author={Kumaar, Saumya and Lyu, Ye and Nex, Francesco and Yang, Michael Ying},
  booktitle={2021 IEEE International Conference on Robotics and Automation (ICRA)},
  title={CABiNet: Efficient Context Aggregation Network for Low-Latency Semantic Segmentation},
  year={2021},
  pages={13517-13524},
  doi={10.1109/ICRA48506.2021.9560977}
}

@article{Kumaar_Real-time_Semantic_Segmentation_2021,
author = {Kumaar, Saumya and Lyu, Ye and Nex, Francesco and Yang, Michael Ying},
doi = {10.1016/j.isprsjprs.2021.06.006},
journal = {ISPRS Journal of Photogrammetry and Remote Sensing},
pages = {124--134},
title = {{Real-time Semantic Segmentation with Context Aggregation Network}},
url = {https://www.sciencedirect.com/science/article/pii/S0924271621001647},
volume = {178},
year = {2021}
}

@article{jocher2026ultralytics,
  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}
}

@software{cabinet_uavid_benchmark,
  author = {Kumaar, Saumya},
  title = {CABiNet: Semantic Segmentation Benchmarking on UAVid (CABiNet vs. YOLO26)},
  url = {https://github.com/dronefreak/CABiNet},
  year = {2026}
}