search

NVlabs / stylegan2-ada-pytorch

StyleGAN2-ADA - Official PyTorch implementation
https://arxiv.org/abs/2006.06676
Other
4.13k stars 1.17k forks source link

Training FFHQR dataset using FFHQ dataset as input #307

Open luser350 opened 5 months ago

luser350 commented 5 months ago

Hi, thanks for sharing your great work. I want to train a stylegan2-ada to take input from an FFHQ image and generate an FFHQR image. The FFHQR dataset is a retouched version of FFHQ.

To my understanding, I can train a FFHQR model using the following command:

python train.py --outdir=training-runs --data=ffhqr/  --gpus=1 --resume=ffhq1024 --aug=noaug

However, this model will learn to generate FFHQR images from noise. The different images can be produced by using a different seed with generate.py

So how can I use stylegan2-ada-pytorch for such img2img translation problem? Your few words can help me a lot. Thanks!!!