search

vanhuyz / CycleGAN-TensorFlow

An implementation of CycleGan using TensorFlow
MIT License
1.19k stars 436 forks source link

Reconstruction works alright but results are poor #96

Open ashutoshbsathe opened 5 years ago

ashutoshbsathe commented 5 years ago

Hi @vanhuyz,

Thanks for the implementation. I'm working on transforming gameplay of "Prince of Persia 1" to "Prince of Persia 2". I have collected 8000+ images of both the games. I've been training for 2400+ steps right now and the results are as shown below - Imgur

Is this a case of model collapse ? If so how should I overcome this issue ?

Thanks again.

[Edit : I see original paper mentioning identity loss for overcoming some issues in the training. I also see that you have identity loss branch on this repo, but your commit for adding identity loss does not include identity loss in optimizer, am I correct ? Where should I add identity loss exactly ?]

gunpowder1473 commented 5 years ago

I add it to the generator loss which used to train two generators. Improved the results in my implementation Hope helpful