https://github.com/renan-siqueira/my-own-began-implementation
My own implementation of a GAN in the BEGAN architecture with pytorch
https://github.com/renan-siqueira/my-own-began-implementation
ai deep-learning discriminator gan generative-adversarial-network generator image image-manipulation interpolate machine-learning neural-networks pytorch training upscale video
Last synced: 6 months ago
JSON representation
My own implementation of a GAN in the BEGAN architecture with pytorch
- Host: GitHub
- URL: https://github.com/renan-siqueira/my-own-began-implementation
- Owner: renan-siqueira
- License: mit
- Created: 2023-09-25T22:23:14.000Z (about 2 years ago)
- Default Branch: main
- Last Pushed: 2023-11-28T21:22:27.000Z (almost 2 years ago)
- Last Synced: 2024-01-27T08:34:25.659Z (over 1 year ago)
- Topics: ai, deep-learning, discriminator, gan, generative-adversarial-network, generator, image, image-manipulation, interpolate, machine-learning, neural-networks, pytorch, training, upscale, video
- Language: Python
- Homepage:
- Size: 84 KB
- Stars: 1
- Watchers: 2
- Forks: 0
- Open Issues: 1
-
Metadata Files:
- Readme: Readme.md
- License: LICENSE
Awesome Lists containing this project
README
# My Own BEGAN Implementation
---
## About
This project implements the Boundary Equilibrium Generative Adversarial Network (BEGAN) architecture. BEGAN is a type of Generative Adversarial Network (GAN) that aims to balance the generator and discriminator during training. [Learn more here](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/00.-About-BEGAN).
---
## Advantages of Using this Project
- **Flexible Image Dimensions:** The architecture can dynamically adjust and train based on the input image size, enabling versatile usage across different datasets without the need for excessive pre-processing.
- **Upscaling Module:** Notably, this project comes equipped with a robust upscaling module. You have the flexibility to train your model in low resolution, optimizing the training time. Once done, employ the upscaling utility to enhance both image and video resolutions. This unique feature not only streamlines the process but also optimizes resource utilization and efficiency.
- **Robust Codebase:** Built with stability and efficiency in mind, the code is resilient against common pitfalls and challenges in GAN training, ensuring consistent and reliable performance.
- **Comprehensive Tool:** Unlike many other GAN codes, this isn't just for training a GAN. This project covers the entire GAN utilization spectrum – from training to image generation and even video generation.
- **Intuitive Use:** Whether you're a GAN veteran or just starting out, the project is designed to be user-friendly. Command-line arguments make operations straightforward, allowing easy toggling between training, image generation, and video creation.
- **Detailed Documentation:** Along with this README, a well-structured Wiki provides deep insights into the Generator and Discriminator modules. This ensures you always have a reference to understand the underlying mechanics.
---
## Documentation
Explore deeper! For a comprehensive breakdown of the project, its modules, setup, and unique features, please refer to our detailed [Wiki](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki). Dive into the specifics of the [Generator Architecture](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/10.-Generator-Architecture) and [Discriminator Architecture](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/11.-Discriminator-Architecture) to enhance your understanding and utilization of the project.
---
## Getting Started
- [Cloning the Repository](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/01.-Cloning-the-Repository)
- [Setting up the Virtual Environment](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/02.-Setting-Up-the-Virtual-Environment)
- [Installing Dependencies](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/03.-Installing-Dependencies)
- [Configuring the GPU](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/04.-Configuring-the-GPU)
- [Preparing the Dataset](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/05.-Preparing-the-Dataset)
- [Configuring Training Parameters](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/06.-Configuring-Training-Parameters)
- [Training the Model](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/07.-Training-the-Model)
- [Generating Images](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/08.-Generating-Images-with-the-Trained-Model)
- [Generating Videos](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/09.-Generating-Videos-Through-Image-Interpolation)
---
## License
This project is open-sourced and available to everyone under the [MIT License](LICENSE).
For more details check the [License and Copyright](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/License-and-Copyright) wiki page.
---
## Contributing
Contributions are welcome! Feel free to open an issue or submit a pull request if you find any bugs or have suggestions for improvements.
Consult the [Contribution and Community Guidelines](https://github.com/renan-siqueira/my-own-BEGAN-implementation/wiki/Contribution-and-Community-Guidelines) wiki page.