/PSN

implement of CVPR 2022 paper《Exploring Endogenous Shift for Cross-domain Detection: A Large-scale Benchmark and Perturbation Suppression Network》

Primary LanguagePythonApache License 2.0Apache-2.0

EDS dataset and the code implementation of PSN

CVPR 2022 paper 《Exploring Endogenous Shift for Cross-domain Detection: A Large-scale Benchmark and Perturbation Suppression Network

Download Link of EDS Dataset:

Please go to the webpage and download according to the prompts.

Prerequisites

  • Python 3.6
  • Pytorch 0.4.1
  • CUDA 8.0 or higher

Compile

pip install -r requirements.txt
cd lib
sh make.sh

Training

The scripts folder has all the training scripts. For example, if you want to train an experiment from domain1 to domain2, just run:

sh scripts/train-1-2-fc.sh

Testing

The scripts folder has all the testing scripts. For example, if you want to test a model trained from domain1 to domain2, just run:

sh scripts/test-all-1-2.sh

Citation

If this work helps your research, please cite the following paper.

@inproceedings{Tao:CVPR22,
  author    = {Renshuai Tao and Hainan Li and Tianbo Wang and Yanlu Wei and Yifu Ding and Bowei Jin and, Hongping Zhi and Xianglong Liu and Aishan Liu},
  title     = {Exploring Endogenous Shift for Cross-domain Detection: A Large-scale Benchmark and Perturbation Suppression Network},
  booktitle = {IEEE CVPR},
  year      = {2022},
  }