tensorflow / gan

Tooling for GANs in TensorFlow
Apache License 2.0
927 stars 246 forks source link

Pix2Pix? #43

Open rachthree opened 3 years ago

rachthree commented 3 years ago

Hello, hope you're staying safe and healthy! I believe back when gan used to be under tensorflow/models/research, it supported pix2pix directly. Was there a reason why it was removed? My group was reliant on it and I'm migrating our framework from TF1 to TF2, and we didn't catch this change.

However, I see in the legacy TF1 code that gan.pix2pix used slim.nets.cyclegan.cyclegan_generator_resnet and slim.nets.pix2pix.pix2pix_discriminator. If we want to reimplement the original that's compatible with TF2, would we use tensorflow_gan.examples.cyclegan.generator.cyclegan_generator_resnet and tensorflow_gan.examples.cyclegan.discriminator.pix2pix_discriminator as it is now, define the loss and optimizer as before in gan.pix2pix, and use tensorflow_gan.estimator.GANEstimator? I'm not sure what might no longer be supported.

But if you happen to have the previous implementation of pix2pix readily available and compatible with tensorflow_gan/TF2, that would be greatly appreciated :) Thanks!