Yume is a project for image generation using Diffusion model implemented in PyTorch.
- Diffusion model for image generation.
- Separate scripts for training and generating images.
- Easy-to-use command-line interface.
-
Clone the repository:
git clone https://github.com/zaibutcooler/yume.git cd yume
-
Install dependencies:
pip install -r requirements.txt
To train the GAN model, use the following command:
```bash
yume-train
```
To generate images with the trained model, use the following command:
```bash
yume-generate
```
yume/
: 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 yume package.
Contributions are welcome! Please follow the Contribution Guidelines.
This project is licensed under the MIT License - see the LICENSE file for details.
Mention any contributors or libraries that you used or were inspired by.
- Zai
- zaiyellyintaung@gmail.com
- Project Link: https://github.com/zaibutcooler/yume