This repo is a PyTorch implementation of applying DDB (Deliberated Domain Bridging for Domain Adaptive Semantic Segmentation) to semantic segmentation. The code is based on mmsegmentaion.
More details can be found in Deliberated Domain Bridging for Domain Adaptive Semantic Segmentation.
In this project, we use python 3.8.13 and pytorch==1.8.1, torchvision==0.9.1, mmcv-full==1.5.0, mmseg==0.22.1 Please refer to get_started.md for install mmsegmentation and mmcv(recommend for 1.5.0)
If your device has internet access, you could set up as follows:
conda create -n dass python=3.8
pip install -r requirements.txt -f https://download.pytorch.org/whl/torch_stable.html
config | train dataset | validation dataset | mIoU |
---|---|---|---|
weights/gta+syn2cs/r2-ckd-pro-bs1x4/weight.pth | gta | cityscape | 62.71 |
weights/gta+syn2cs/r2-ckd-pro-bs1x4/weight.pth | gta+syn | cityscape | 68.99 |
weights/gta2cs+map/s2-ckd-pro-bs1x4/weight.pth | gta gta |
cityscape mapillary |
60.38 56.85 |
The above weight and log can be obtained through BaiduYun. After downloading, please put it under the project folder
Cityscapes: Please, download leftImg8bit_trainvaltest.zip and
gt_trainvaltest.zip from here
and extract them to data/cityscapes
.
mapillary Please, download MAPILLARY v1.2 from here
GTA: Please, download all image and label packages from
here and extract
Synthia: Please, download SYNTHIA-RAND-CITYSCAPES from
here and extract it to data/synthia
.
them to data/gta
.
Then, you should prepare data as follows:
cd DASS
mkdir data
# If you prepare the data at the first time, you should convert the data for validation
python tools/convert_datasets/gta.py data/gta/ # Source domain
python tools/convert_datasets/synthia.py data/synthia/ # Source domain
python tools/convert_datasets/synscapes.py data/synscapes/ # Source domain
# convert mapillary to cityscape format and resize it for efficient validation
python tools/convert_datasets/mapillary2cityscape.py data/mapillary/ \
data/mapillary/cityscape_trainIdLabel --train_id # Source domain
python tools/convert_datasets/mapillary_resize.py data/mapillary/validation/images \
data/mapillary/cityscape_trainIdLabel/val/label data/mapillary/half/val_img \
data/mapillary/half/val_label
The final folder structure should look like this:
DASS
├── ...
├── weights
├── data
│ ├── cityscapes
│ │ ├── leftImg8bit
│ │ │ ├── train
│ │ │ ├── val
│ │ ├── gtFine
│ │ │ ├── train
│ │ │ ├── val
│ ├── mapillary
│ │ ├── training
│ │ ├── cityscape_trainIdLabel
│ │ ├── half
│ │ | ├── val_img
│ │ | ├── val_label
├── ...
Download the folder weights and place it in the project directory
Verify by selecting the different config files in configs/tests
python tools/test.py {config} {weight} --eval mIoU
Using following commands, you will receive two complementary teacher models (cu_model and ca_model)
# Train on the region-path (using cut-mix for domain bridging)
python tools/train.py configs/gtav2cityscapes/r1_st_cu_dlv2_r101v1c_1x4_512x512_40k_gtav2cityscapes.py
# Train on the class-path (using class-mix for domain bridging)
python tools/train.py configs/gtav2cityscapes/r1_st_ca_dlv2_r101v1c_1x4_512x512_40k_gtav2cityscapes.py
# Train on the region-path (using cut-mix for domain bridging) (Train with multiple GPUs)
bash tools/dist_train.sh configs/gtav2cityscapes/r1_st_ca_dlv2_r101v1c_2x2_512x512_40k_gtav2cityscapes.py ${GPU_NUM}
# Train on the class-path (using class-mix for domain bridging) (Train with multiple GPUs)
bash tools/dist_train.sh configs/gtav2cityscapes/r1_st_cu_dlv2_r101v1c_2x2_512x512_40k_gtav2cityscapes.py ${GPU_NUM}
If you want to generate prototypes for rectifying the pseudo label produced in Step 2. You should run:
# Generating prototypes for the region-path teacher on target domain
python tools/cal_prototypes/cal_prototype.py {CU_MODEL_CONFIG_DIR} --checkpoint={CU_MODEL_CHECKPOINT_DIR}
# Generating prototypes for the region-path teacher on target domain
python tools/cal_prototypes/cal_prototype.py {CA_MODEL_CONFIG_DIR} --checkpoint={CA_MODEL_CHECKPOINT_DIR}
After step 1, you should rename the checkpoints and put them in the checkpoints' folder manually. Such as:
DASS
├── ...
├── checkpoints
│ ├── gta2cs_stage1
│ │ ├── gta2cs_st-cu_dlv2.pth
│ │ ├── gta2cs_st-ca_dlv2.pth
├── ...
Then, you can run the following command for Cross-path Knowledge Aggregation:
# Distillate the knowledge from two teacher models to a student model
python tools/train.py configs/gtav2cityscapes/r1_ckd_dlv2_r101v1c_1x4_512x512_40k_gtav2cityscapes.py
# Train with multiple GPUs
bash tools/dist_train.sh configs/gtav2cityscapes/r1_ckd_dlv2_r101v1c_2x2_512x512_40k_gtav2cityscapes.py ${GPU_NUM}
# Self-training again with weights initialized by step2 on stage 1
python tools/train.py configs/uda/st/gta2cs_st-cu-r2_dlv2red-adapter_r101v1c_poly10warm_s0.py
# Self-training again with weights initialized by step2 on stage 1
python tools/train.py configs/uda/st/gta2cs_st-ca-r2_dlv2red-adapter_r101v1c_poly10warm_s0.py