Skip to content

Implementation of DualGAN for general-purpose image-to-image translation done as an assignment for the course BITS-F312 (Neural Networks and Fuzzy Logic)

Notifications You must be signed in to change notification settings

Senpai1199/dualGAN

Repository files navigation

dualGAN

Tensorflow implementation of the paper: DualGAN: Unsupervised Dual Learning for Image-to-Image Translation
Link to Complete Dataset

Folders in the repository:

  1. datasets : Contains 2 datasets namely "facades.zip" and "day-night.zip" which were used to train the model.
  2. results : Contains the results obtained from our implementation for each of the above dataset.
    2.1 Receptive fields used for the day-night dataset: 70x70, 16x16, 1x1
    2.2 Receptive fields used for the facades dataset: 70x70

Individual files:

  1. training_parameters.json : Contains the parameters for training the model such as no. of epochs, dataset_name, batch size, directory paths.
  2. tf1_dualgan.py : File containing the core dual GAN architecture and the whole logic for generators, discriminators, training, testing.
  3. helper.py : Contains the Dataset class with helper functions such as loading and fetching images.
  4. calc_receptive_field.py : Contains logic for calculating patch sizes for different receptive field sizes.

About

Implementation of DualGAN for general-purpose image-to-image translation done as an assignment for the course BITS-F312 (Neural Networks and Fuzzy Logic)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages