liuguoyou / Generalized-One-shot-GAN-Adaptation

Generalized One-shot Domain Adaptation of Generative Adversarial Networks (NeurIPS 2022)

Quick start

We provide some pre-trained models and a inference demo on Google Colab to reproduce the qualitative results in paper. It also includes the command to train the model.

Usage

• Please refer to the colab to create the envs and download the source models. The envs mainly includes StyleGAN and CLIP dependencies, and some commonly used packages like cv2.
• We have prepared the scripts to train new models
  • For training model without entities, sh scripts/train_OSGA.sh
  • For training model with entities, sh scripts/train_GOGA.sh
  • For inference, sh scripts/inference.sh

Notation

This project has not been well cleaned and may contain some redundant files. If you have questions about the codes or our paper, feel free to submit issues or contact with me.

This project heavily depends on the code of NVIDIA StyleGAN v3, hence please follow its licenses.

Supplementary.pdf

Citation

@article{Zhang2022GeneralizedOD,
  title={Generalized One-shot Domain Adaptation of Generative Adversarial Networks},
  author={Zicheng Zhang and Yinglu Liu and Congying Han and Tiande Guo and Ting Yao and Tao Mei},
  journal={arXiv preprint arXiv:2209.03665},
  year={2022}
}