search

mingyuliutw / UNIT

Unsupervised Image-to-Image Translation
Other
1.98k stars 360 forks source link

Identity Loss #100

Closed mhusseinsh closed 5 years ago

mhusseinsh commented 5 years ago

Hello,

One of the interesting parts in the CycleGAN implementation is the identity loss, which ensures that the network should act as an identity mapping not translation in case if it is fed from a real sample from the target domain. Example: Snow to Summer translation if the network is fed with Snow, it should output a summer image if it is fed with a summer image, "using the same Encoder-Decoder for the Snow2Summer", it should output the exact identical summer image with no translation

I wonder, if the UNIT is able to handle such scenario, or if there is something similar to the identity loss is implemented.

mingyuliutw commented 5 years ago

The Cycle Consistency loss is not always desirable in my experiences. But yes, we have the cycle consistency loss in our implementation, which can be enabled by setting the weights to non-zero. Please check the example yaml file.