Skip to content

LoyoYang/DeCoTa

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

67 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation



Image

Requirements

The code is developed under Python 3.6.5 and PyTorch 1.4.0

To install, pip install -r requirements.txt


Prepare dataset

To reproduce the DomainNet results, download DomainNet from http://ai.bu.edu/M3SDA/ following the instructions on the page.

Your dataset root is expected to contain folders named after all the domains, for example:

PATH-TO-DATASET-ROOT/clipart


Train your own model

There are seven adaptation scenarios in the DomainNet experiment.

Specify the Source and Target domain by either

--source X --target Y or --st X_Y_index

Example to reproduce the DomainNet 3-shot result, from Real to Clipart (X_Y_index=1), saving checkpoint:

python main.py --root PATH-TO-DATASET-ROOT/ --st 1 --save_check

Results will be saved at ./record/multi/mico Checkpoint will be saved at ./checkpoint


Eval

Example to evaluate a saved model at iteration 20000 from ./checkpoint, from Real to Clipart:

python main.py --root PATH-TO-DATASET-ROOT/ --st 1 --eval --net_resume Net_iter_model_mico_real_to_clipart_step_20000.pth.tar

About

ICCV 2021: Deep Co-Training with Task Decomposition for Semi-supervised Domain Adaptation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages