/DASS

An official PyTorch implementation of "Bi-directional Contrastive Learning for Domain Adaptive Semantic Segmentation", ECCV 2022.

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

[ECCV22] Bi-directional Contrastive Learning for Domain Adaptive Semantic Segmentation

This is a PyTorch implementation of [Bi-directional Contrastive Learning for Domain adaptive Semantic Segmentation].

DASS

Requirements

To install requirements:

  • Python 3.6
  • Pytorch 1.4.0

Getting Started

  1. Download the dataset.
  2. Download the ImageNet-pretrained Model [Link].

Training

Train the source-only model:

python so_run.py

Train our model:

python run.py

Evaluation

To perform evaluation on single model:

python eval.py --frm model.pth --single