Skip to content

UTSAVS26/GAN-Models

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GAN Models Repository


Welcome to the GAN Models repository! This repository contains implementations of various Generative Adversarial Network (GAN) models using PyTorch. Each GAN model is implemented in its own folder within this repository.


Implemented GAN Models

  • Description: BasicGAN is a simple implementation of the traditional Generative Adversarial Network (GAN) architecture.
  • Folder: BasicGAN/
  • Files:
    • BasicGAN.py: Implementation of the Basic GAN model.
    • test_BasicGAN.py: Script to generate sample images using the trained Basic GAN model.
    • README.md: Detailed information about BasicGAN including usage instructions, parameters, and examples.
  • Description: ACGAN extends the GAN architecture by incorporating class information into both the generator and the discriminator.
  • Folder: ACGAN/
  • Files:
    • ACGAN.py: Implementation of the ACGAN model.
    • test_ACGAN.py: Script to generate sample images using the trained ACGAN model.
    • README.md: Detailed information about ACGAN including usage instructions, parameters, and examples.
  • Description: EBGAN focuses on matching the energy distribution of generated samples to that of real data.
  • Folder: EBGAN/
  • Files:
    • EBGAN.py: Implementation of the EBGAN model.
    • test_EBGAN.py: Script to generate sample images using the trained EBGAN model.
    • README.md: Detailed information about EBGAN including usage instructions, parameters, and examples.
  • Description: cGAN conditions both the generator and discriminator on additional information to generate more specific outputs.
  • Folder: cGAN/
  • Files:
    • cGAN.py: Implementation of the cGAN model.
    • test_cGAN.py: Script to generate sample images using the trained cGAN model.
    • README.md: Detailed information about cGAN including usage instructions, parameters, and examples.
  • Description: InfoGAN learns interpretable and disentangled representations using unsupervised learning.
  • Folder: InfoGAN/
  • Files:
    • InfoGAN.py: Implementation of the InfoGAN model.
    • test_InfoGAN.py: Script to generate sample images using the trained InfoGAN model.
    • README.md: Detailed information about InfoGAN including usage instructions, parameters, and examples.

Getting Started

To use any of the implemented GAN models, follow these general steps:

  1. Clone the repository:

    git clone https://github.com/UTSAVS26/GAN-models.git
    cd GAN-models
  2. Navigate to the specific GAN model folder (e.g., BasicGAN/, ACGAN/, etc.).

  3. Install dependencies: Each model folder contains a requirements.txt file listing necessary dependencies. Install them using pip:

    pip install -r requirements.txt
  4. Train the GAN model: Run the corresponding Python script (e.g., BasicGAN.py, ACGAN.py, etc.) to train the GAN model on your dataset.

  5. Generate new images: After training, use the provided test script (e.g., test_BasicGAN.py, test_ACGAN.py, etc.) to generate new sample images.


Contributing

Contributions to this repository are welcome! If you have ideas for improvements, additional GAN models, or better implementations, feel free to open an issue or submit a pull request.

Author


Happy Generating with GAN Models! 🖼️

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages