ZhangJUJU / OPDA_BP

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

A Pytorch Implementation of Open Set Domain Adaptation by Back-propagation (ECCV 2018)

Introduction

Official Implementation of Open Set Domain Adaptation by Back-propagation. We publicize code for VisDA experiment.

Data Preparation

Follow https://github.com/VisionLearningGroup/taskcv-2017-public/tree/master/classification to download visda dataset. Input path to utils/list_visda.py and run list_visda.py to generate path list file.

Train

 CUDA_VISIBLE_DEVICES=$GPU_ID python trainer_osda.py \
                    --net alex

Note

We reported the performance using pytorch 0.3. Results below are obtained using pytorch 0.4 for Alexnet and VGG. We are investigating the issues caused by the chanage in versions.

Network bicycle bus car motorcycle train truck unknown
AlexNet 54.3 76.0 38.6 77.8 71.7 1.0 70.6
VGG 58.4 69.6 50.0 81.3 81.2 28.3 91.7

Citation

Please cite the following reference if you utilize this repository for your project.

@inproceedings{saito2018open,
  title={Open set domain adaptation by backpropagation},
  author={Saito, Kuniaki and Yamamoto, Shohei and Ushiku, Yoshitaka and Harada, Tatsuya},
  booktitle={Proceedings of the European Conference on Computer Vision (ECCV)},
  pages={153--168},
  year={2018}
}

About


Languages

Language:Python 100.0%