Skip to content

gouayao/CSR

Repository files navigation

#CSR

This code implements CSR model, described in the paper Rethinking Cross-domain Semantic Relation for Few-shot Image Generation, Yao Gou, Min Li, Yilong Lv, Yusen Zhang, Yuhang Xing, APIN, 2023.

##The code borrows heavily from the PyTorch implementation of CDC: https://github.com/utkarshojha/few-shot-gan-adaptation

##Our CSR is inspired by ContraD https://github.com/jh-jeong/ContraD

##Train the CSR model:

python train.py  --ckpt ./checkpoints/source_ffhq.pt --data_path ./processed_data/10-shot/ --exp 10-shot_CSR

Acknowledgment

As mentioned before, the StyleGAN2 model is borrowed from this wonderful pytorch implementation by @rosinality. We are also thankful to @mseitzer and @richzhang for their user friendly implementations of computing FID score and LPIPS metric respectively.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages