search

clovaai / stargan-v2

StarGAN v2 - Official PyTorch Implementation (CVPR 2020)
Other
3.5k stars 658 forks source link

How to train stargan-v2 on my own dataset? #148

Open k2sq opened 1 year ago

k2sq commented 1 year ago

What should my folder structure look like? Do I have to change some code? Really appreciated your work.

inhyukpark2 commented 1 year ago

python main.py --mode train --num_domains 2 --w_hpf 1 \ --lambda_reg 1 --lambda_sty 1 --lambda_ds 1 --lambda_cyc 1 \ --train_img_dir data/celeba_hq/train \ --val_img_dir data/celeba_hq/val

You can change train, val directory, hyperparameter.

study-clever commented 11 months ago

我的文件夹结构应该是什么样子的？我必须更改一些代码吗？真的很感谢你的工作。

What should my folder structure look like? Do I have to change some code? Really appreciated your work.

Have you implemented training your own dataset? Can you exchange ideas?