Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

HRDA: Context-Aware High-Resolution Domain-Adaptive Semantic Segmentation

About

Unsupervised domain adaptation (UDA) aims to adapt a model trained on the source domain (e.g. synthetic data) to the target domain (e.g. real-world data) without requiring further annotations on the target domain. This work focuses on UDA for semantic segmentation as real-world pixel-wise annotations are particularly expensive to acquire. As UDA methods for semantic segmentation are usually GPU memory intensive, most previous methods operate only on downscaled images. We question this design as low-resolution predictions often fail to preserve fine details. The alternative of training with random crops of high-resolution images alleviates this problem but falls short in capturing long-range, domain-robust context information. Therefore, we propose HRDA, a multi-resolution training approach for UDA, that combines the strengths of small high-resolution crops to preserve fine segmentation details and large low-resolution crops to capture long-range context dependencies with a learned scale attention, while maintaining a manageable GPU memory footprint. HRDA enables adapting small objects and preserving fine segmentation details. It significantly improves the state-of-the-art performance by 5.5 mIoU for GTA-to-Cityscapes and 4.9 mIoU for Synthia-to-Cityscapes, resulting in unprecedented 73.8 and 65.8 mIoU, respectively. The implementation is available at https://github.com/lhoyer/HRDA.

Lukas Hoyer, Dengxin Dai, Luc Van Gool• 2022

Related benchmarks

TaskDatasetResultRank
Semantic segmentationGTA5 → Cityscapes (val)
mIoU73.8
586
Semantic segmentationSYNTHIA to Cityscapes (val)
Rider IoU68.9
480
Semantic segmentationSYNTHIA to Cityscapes
Road IoU85.2
159
Semantic segmentationMapillary (val)
mIoU61.2
153
Semantic segmentationCityscapes adaptation from Synthia 1.0 (val)
Person IoU79.4
114
Semantic segmentationGTA5 to Cityscapes 1.0 (val)
Road IoU96.4
98
Semantic segmentationBDD100K (val)
mIoU49.1
84
Semantic segmentationACDC (test)
mIoU67.96
74
Semantic segmentationGTA to Cityscapes
Road IoU96.4
72
Semantic segmentationCityscapes to ACDC (test)
mIoU68
67
Showing 10 of 58 rows

Other info

Code

Follow for update