Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add pytorch_Pix2Pix_cGAN.py: implementation of Pix2Pix with conditional GAN (cGAN) #14

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

anh-nn01
Copy link

Primary contributions:

  1. Add implementation for Pix2Pix using cGAN, a general framework for image translation tasks (e.g: Satellite to Maps, Image to Anime, Scene Day to Scene Night, etc.). Reference paper: https://arxiv.org/abs/1611.07004.
  2. Add implementation of U-Net model for Pix2Pix's Generator.
  3. Add implementation of CNN Binary Classifier for Pix2Pix Discriminator.
  4. Add implementation of the cGAN training framework consisting of U-Net Generator and CNN Discriminator.
  5. Modified checkpoint saving mechanism: saving checkpoint every 5 epochs.
  6. Provided hyperparameters for training Pix2Pix, specifically batch_size, learning_rate, num_epochs, beta_1 and beta_2 (in Adam Optimizer), L1 regularizer, and number of GPUs.

IMPORTANT NOTE (for README.md):

  1. To test the new Pix2Pix framework, we test the model on the Satellite Imagery to Map the translation task.
  2. The task dataset should be downloaded on the same level of pytorch_Pix2Pix_cGAN.py.
  3. The dataset can be downloaded here .
  4. If you want to try the model on another task, such as Image to Sketch, please download the dataset and change the dataset directory in pytorch_Pix2Pix_cGAN.py.

Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please ignore this file in this commit and only care about requirements.txt. The correct pytorch_Pix2Pix_cGAN is in the later commit.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant