CycleGAN icon indicating copy to clipboard operation
CycleGAN copied to clipboard

cyclegan training

Open suke27 opened this issue 6 years ago • 3 comments

Hi, I have a problem for cycle gan training, currently our training dataset need many image(both A and B), can I do art style transfer based just one image. like Gatys do, do art style transfer for any image using any style image do you try it?

suke27 avatar Feb 13 '18 08:02 suke27

We haven't tried it. It is worth trying. I can imagine there might be overfitting issue as the discriminator only needs to memorize one image.

junyanz avatar Feb 15 '18 22:02 junyanz

i really try, but not good result, another issue, I found GAN result always introduce some abnormal feature, you can check attached file, is it GAN common issue, how do we resolve it. I try filter training images(remove images which include watermark), but useless, @junyanz Thank you!

2b3g9v 2b3g9v_style

suke27 avatar Feb 22 '18 03:02 suke27

It's hard to train a GAN with only one target image. One thing you can try is to use a smaller patch size. Let's assume that your image is 512x512. You can set loadSize=512, and fineSize=64 or 128 during training, and set loadSize=512, fineSize=512 during test.

junyanz avatar Feb 22 '18 16:02 junyanz