Skip to content
/ SM-PPM Public

(AAAI2022) Style Mixing and Patchwise Prototypical Matching for One-Shot Unsupervised Domain Adaptive Semantic Segmentation

Notifications You must be signed in to change notification settings

W-zx-Y/SM-PPM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SM-PPM

This is a Pytorch implementation of our paper "Style Mixing and Patchwise Prototypical Matching for One-Shot Unsupervised Domain Adaptive Semantic Segmentation".

AAAI2022 [arxiv]

Requirements

  • python3.7
  • pytorch>=1.5.0
  • cuda10.2

Datasets

GTA5

Synthia

Cityscapes

Dark Zurich

Pretrained Models

  1. The source only model for GTA5 and Synthia are provided by AdaptSegNet.
  2. For day-to-night adaptation, please download the model pretrained on Cityscapes here.

Download these pretrained models and put them into the pretrained_model folder.

Training and Evaluation Instruction for GTA5->Cityscapes

Modify the all data paths in the train_config.py and test_config.py.

bash run.sh

Acknowledgment

Part of our code is from MixStyle and AdaptSegNet. We gratefully thank the authors for their great work. Also thank the authors of ASM for introducing this one-shot UDA setting.

Citation

If you think this paper is useful for your research, please cite our paper:

@inproceedings{wu2021style,
  title={Style Mixing and Patchwise Prototypical Matching for One-Shot Unsupervised Domain Adaptive Semantic Segmentation},
  author={Wu, Xinyi and Wu, Zhenyao and Lu, Yuhang and Ju, Lili and Wang, Song},
  booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
  year={2022}
}

About

(AAAI2022) Style Mixing and Patchwise Prototypical Matching for One-Shot Unsupervised Domain Adaptive Semantic Segmentation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published