This repository contains implementations of various Generative Adversarial Networks (GANs) as part of the Generative AI course assignment. The projects cover image generation, image-to-image translation, and domain translation.
The project is divided into three main parts:
- Objective: Generate realistic anime faces from random noise.
- Model: Deep Convolutional Generative Adversarial Network (DCGAN).
- Dataset: Anime Faces dataset.
- Framework: PyTorch.
- Objective: Convert black-and-white anime sketches into fully colored images.
- Model: Pix2Pix (Conditional GAN).
- Dataset: Anime sketch-to-photo pairs.
- Key Features: Uses a U-Net generator and a PatchGAN discriminator.
- Objective: Unpaired image-to-image translation between the sketch domain and the photo domain.
- Model: Cycle-Consistent Adversarial Network (CycleGAN).
- Dataset: TU-Berlin sketches and custom photo collections.
- Key Features: Implements cycle-consistency loss and identity loss for stable translation without paired training data.
- Python
- PyTorch
- Torchvision
- Matplotlib
- NumPy
- Kaggle T4 GPUs (for training)
- Email: wareeshaashraf09@gmail.com
- Clone the repository:
git clone https://github.com/wareeshayy/Gen_AI.git
- Open the
.ipynbfiles in Google Colab or Kaggle. - Ensure you have the required datasets or update the paths within the notebooks.
- Run the cells to train or evaluate the models.