SAM-Enhanced Segmentation on Road Datasets: Balancing Critical Classes in Autonomous Driving
Paper in proceeding, 2026

Dense semantic segmentation is essential for autonomous driving, yet many multi-modal datasets lack pixel-level annotations. The Zenseact Open Dataset (ZOD) provides rich multi-sensor data but only bounding-box labels, limiting its use for segmentation research. Our primary contribution is a Segment Anything Model (SAM)-based annotation pipeline that produces dense, pixel-level annotations for ZOD by converting bounding boxes into semantic masks. In this pilot study, we process over 100,000 frames and manually curate a 2,300-frame subset (36% acceptance rate) to establish a reliable baseline. Using these annotations, we evaluate transformer-based CLFT and CNN-based DeepLabV3+ architectures across diverse weather conditions, achieving up to 48.1% mIoU with CLFT-Hybrid. To address extreme class imbalance, where pedestrians, cyclists, and signs constitute less than 1% of pixels, we explore specialized models targeting rare classes. We further validate the pipeline on the Iseauto autonomous-vehicle platform, achieving 77.5% mIoU, and show that SAM-derived representations transfer effectively across sensor configurations via bidirectional transfer learning. All code and annotations are released to support reproducible research.

DeepLabV3+

autonomous driving

class imbalance

model specialization

Vision Transformers

semantic segmentation

computational efficiency

multi-modal fusion

Segment Anything Model

transfer learning

Author

Toomas Tahves

Tallinn University of Technology (TalTech)

Mauro Bellone

Universitas Mercatorum

Tallinn University of Technology (TalTech)

Junyi Claude Gu

University of Gothenburg

Chalmers, Computer Science and Engineering (Chalmers), Interaction Design and Software Engineering

Raivo Sell

Tallinn University of Technology (TalTech)

IEEE ASME International Conference on Advanced Intelligent Mechatronics AIM

21596247 (ISSN) 21596255 (eISSN)


9798319536112 (ISBN)

2026 IEEE/ASME International Conference on Advanced Intelligent Mechatronics, AIM 2026
Genova, Italy,

Subject Categories (SSIF 2025)

Computer graphics and computer vision

DOI

10.1109/AIM65483.2026.11658218

More information

Latest update

9/11/2026