Skip to content
/ UNITE Public
forked from fnzhan/UNITE

Unbalanced Feature Transport for Exemplar-based Image Translation (CVPR 2021)

Notifications You must be signed in to change notification settings

TRYQ1998/UNITE

 
 

Repository files navigation

Unbalanced Feature Transport for Exemplar-based Image Translation (CVPR 2021)

Teaser

Preparation

Clone the Synchronized-BatchNorm-PyTorch repository.

cd models/networks/
git clone https://github.com/vacancy/Synchronized-BatchNorm-PyTorch
cp -rf Synchronized-BatchNorm-PyTorch/sync_batchnorm .
cd ../../

VGG model for computing loss. Download from here, move it to models/.

For the preparation of datasets, please refer to CoCosNet.

Pretrained Models

The pretrained models can be downloaded from here. Saving the pretrained models in checkpoints.

Inference Using Pretrained Model

Then run the command

bash test_ade.sh

Training

Then run the command

bash train_ade.sh

Citation

If you use this code for your research, please cite our papers.

@inproceedings{zhan2021unite,
  title={Unbalanced Feature Transport for Exemplar-based Image Translation},
  author={Zhan, Fangneng and Yu, Yingchen and Cui, Kaiwen and Zhang, Gongjie and Lu, Shijian and Pan, Jianxiong and Zhang, Changgong and Ma, Feiying and Xie, Xuansong and Miao, Chunyan},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={15028--15038},
  year={2021}
}

Acknowledgments

This code borrows heavily from CoCosNet. We also thank SPADE, Synchronized Normalization and Geometric Loss.

About

Unbalanced Feature Transport for Exemplar-based Image Translation (CVPR 2021)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.0%
  • Shell 1.0%