roysubhankar / dwt-domain-adaptation

Code for paper "Unsupervised Domain Adaptation using Feature-Whitening and Consensus Loss" (CVPR 2019)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Domain Whitening Transform for Unsupervised Domain Adaptation (CVPR 2019)

Official PyTorch github repository for the paper Unsupervised Domain Adaptation using Feature-Whitening and Consensus Loss published in The Conference on Computer Vision and Pattern Recognition (CVPR) held at Long Beach, California in June, 2019.

Prerequisites

  • Pytorch 1.0
  • Python 3.5

Usage

  • Office-Home: To run the experiments on the OfficeHome dataset first you need to download the dataset from this page. Following this step, you would need to download the ResNet50 pre-trained checkpoint, trained on ImageNet with the BatchNorm layers (in the first conv layer and the first Res block) replaced by whitening normalization layers. The pre-trained weights is available here.
python resnet50_dwt_mec_officehome.py --s_dset_path path-to-source-dataset-folder --t_dset_path path-to-target-dataset       folder --resnet_path path-to-pre-trained-resnet50-weights
  • USPS -> MNIST:
python usps_mnist.py --group_size 4 --source 'usps' --target 'mnist'

If you find this code useful for your research, please cite our paper:

@article{roy2019unsupervised,
  title={Unsupervised Domain Adaptation using Feature-Whitening and Consensus Loss},
  author={Roy, Subhankar and Siarohin, Aliaksandr and Sangineto, Enver and Bulo, Samuel Rota and Sebe, Nicu and Ricci, Elisa},
  booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
  year={2019}
}

Subhankar's GitHub stats

About

Code for paper "Unsupervised Domain Adaptation using Feature-Whitening and Consensus Loss" (CVPR 2019)


Languages

Language:Python 100.0%