Skip to content

Latest commit

 

History

History
7 lines (5 loc) · 1001 Bytes

README.md

File metadata and controls

7 lines (5 loc) · 1001 Bytes

Learning to program a CycleGAN model using horse2zebra

My first attempt to implement a CycleGAN in TensorFlow based on Jun-Yan Zhu, Taesung Park, Phillip Isola, and Alexei A. Efros. "Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks", in IEEE International Conference on Computer Vision (ICCV), 2017. See: https://junyanz.github.io/CycleGAN/

Current status

The model seems to work, but iterations take quite a long time so it's hard to verify. I do not think the slowness is related to a bug, but it's worth another look. Part of the issue is the very large increase in parameters when the residual block concatenates its input rather than adds to it. The paper does not make it clear which approach they chose (addition vs concatenation). Either way, I think I will need to devote quite a bit of computation time to check how well it works.

My image