/ve-gans

Production level Generative Adversarial Network with PyTorch.

Primary LanguageJupyter NotebookGNU General Public License v3.0GPL-3.0

ve-gans: Image Generation with GANs using PyTorch

Overview

ve-gans is a project for image generation using Generative Adversarial Networks (GANs) implemented in PyTorch.

Features

  • GAN model for image generation.
  • Separate scripts for training and generating images.
  • Easy-to-use command-line interface.

Installation

  1. Clone the repository:

    git clone https://github.com/zaibutcooler/ve-gans.git
cd ve-gans

  2. Install dependencies:

    pip install -r requirements.txt

Usage

Training

To train the GAN model, use the following command:

ve-gans-train

Generating Images

To generate images with the trained model, use the following command:

ve-gans-generate

Project Structure

  • ve_gans/: Python package containing GAN implementation and utilities.
    • generator.py: Implementation of the GAN generator.
    • discriminator.py: Implementation of the GAN discriminator.
    • utils.py: Utility functions.
  • requirements.txt: List of project dependencies.
  • setup.py: Setup script for installing the package.
  • main.py: Example script for using the ve-gans package.

Contributing

Contributions are welcome! Please follow the Contribution Guidelines.

License

This project is licensed under the MIT License - see the LICENSE file for details.

Acknowledgments

Mention any contributors or libraries that you used or were inspired by.

Contact