Fine-tuned YOLO26x semantic segmentation model for aerial drone imagery using the AeroScapes benchmark dataset.
AeroScapes is an aerial semantic segmentation benchmark of ~3,200 720p images captured by a fleet of drones across roughly 10 flight sequences, covering urban and suburban scenes with people, vehicles, and small objects.
1from huggingface_hub import hf_hub_download
2from ultralytics import YOLO
3
4weights = hf_hub_download(
5 repo_id="dronefreak/aeroscapes-yolo26x-sem",
6 filename="best.pt"
7)
8
9model = YOLO(weights)
1results = model.predict(source="image.png", task="semantic", imgsz=896)
2mask = results[0].semantic_mask.cpu().numpy().data # (H, W) class-ID map
Trained with the
CABiNet repository, which pairs its own real-time segmentation trainer with a parallel Ultralytics YOLO26-sem pipeline — shared dataset tooling, training/eval, and mIoU benchmarking across UAVid, AeroScapes, and VDD. Star the repo if you find these models useful!
1@inproceedings{nigam2018ensemble,
2 title={Ensemble Knowledge Transfer for Semantic Segmentation},
3 author={Nigam, Ishan and Huang, Chen and Ramanan, Deva},
4 booktitle={2018 IEEE Winter Conference on Applications of Computer Vision (WACV)},
5 year={2018}
6}
7
8@INPROCEEDINGS{9560977,
9 author={Kumaar, Saumya and Lyu, Ye and Nex, Francesco and Yang, Michael Ying},
10 booktitle={2021 IEEE International Conference on Robotics and Automation (ICRA)},
11 title={CABiNet: Efficient Context Aggregation Network for Low-Latency Semantic Segmentation},
12 year={2021},
13 pages={13517-13524},
14 doi={10.1109/ICRA48506.2021.9560977}
15}
16
17@article{Kumaar_Real-time_Semantic_Segmentation_2021,
18 author = {Kumaar, Saumya and Lyu, Ye and Nex, Francesco and Yang, Michael Ying},
19 doi = {10.1016/j.isprsjprs.2021.06.006},
20 journal = {ISPRS Journal of Photogrammetry and Remote Sensing},
21 pages = {124--134},
22 title = {{Real-time Semantic Segmentation with Context Aggregation Network}},
23 url = {https://www.sciencedirect.com/science/article/pii/S0924271621001647},
24 volume = {178},
25 year = {2021}
26}
27
28@article{jocher2026ultralytics,
29 title={Ultralytics YOLO26: Unified Real-Time End-to-End Vision Models},
30 author={Jocher, Glenn and Qiu, Jing and Liu, Mengyu and Lyu, Shuai and Akyon, Fatih Cagatay and Kalfaoglu, Muhammet Esat},
31 journal={arXiv preprint arXiv:2606.03748},
32 year={2026}
33}
34
35@software{cabinet_uavid_benchmark,
36 author = {Kumaar, Saumya},
37 title = {CABiNet: Semantic Segmentation Benchmarking on UAVid (CABiNet vs. YOLO26)},
38 url = {https://github.com/dronefreak/CABiNet},
39 year = {2026}
40}