https://github.com/deepmancer/diffusion-gan-vae-pytorch

A PyTorch implementation of various deep generative models, including Diffusion (DDPM), GAN, cGAN, and VAE.
https://github.com/deepmancer/diffusion-gan-vae-pytorch

conditional-diffusion conditional-gan conditional-generation diffusion-models fashion-mnist from-scratch generative-adversarial-network mnist pytorch vae-pytorch

Last synced: 3 months ago
JSON representation

A PyTorch implementation of various deep generative models, including Diffusion (DDPM), GAN, cGAN, and VAE.

• Host: GitHub
• URL: https://github.com/deepmancer/diffusion-gan-vae-pytorch
• Owner: deepmancer
• Created: 2023-05-24T12:29:34.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2024-08-16T12:18:00.000Z (6 months ago)
• Last Synced: 2024-10-11T20:02:56.981Z (4 months ago)
• Topics: conditional-diffusion, conditional-gan, conditional-generation, diffusion-models, fashion-mnist, from-scratch, generative-adversarial-network, mnist, pytorch, vae-pytorch
• Language: Jupyter Notebook
• Homepage:
• Size: 20.6 MB
• Stars: 4
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# 🎨 Deep Generative Models in PyTorch



  

  

  



Welcome to the **Diffusion-GAN-VAE-PyTorch** repository! This project contains PyTorch implementations (from scratch) of various deep generative models, including Variational Autoencoders (VAE), Generative Adversarial Networks (GAN), Conditional GANs, Diffusion Models, and Conditional Diffusion Models. Explore the power of these models through code and stunning visualizations!

---

## 🌠 Results

### Fashion MNIST



  



### Conditional Diffusion Model



  

  



### Diffusion Model



  

  



### Conditional GAN



  



### GAN



  

  



### VAE



  

  



---

## 🛠️ Requirements

To get started, ensure you have the following dependencies installed:

- **Python 3**: The language of choice for deep learning.

- **PyTorch**: The core framework for building and training the models.

- **NumPy**: For numerical operations.

- **Matplotlib**: To visualize the results.

- **tqdm**: For progress bars.

You can install these dependencies via pip:

```bash

pip install torch numpy matplotlib tqdm

```

---

## 🔍 Models Overview

### 🔹 Variational Autoencoder (VAE)

The **VAE** is a powerful generative model that learns a latent representation of the input data, which can be used to generate new, unseen data samples. VAEs are particularly useful for unsupervised learning and data generation tasks.

### 🔹 Generative Adversarial Network (GAN)

The **GAN** framework consists of two neural networks—the generator and the discriminator—that compete against each other. The generator creates fake data samples, while the discriminator tries to distinguish between real and fake samples. Through this adversarial process, the generator improves its ability to create realistic data.

### 🔹 Conditional Generative Adversarial Network (cGAN)

The **cGAN** extends the GAN by conditioning the data generation process on additional information, such as class labels. This allows for more controlled generation of images, enabling the creation of samples with specific attributes or belonging to a particular class.

### 🔹 Diffusion Models

**Diffusion Models** are a class of generative models that learn to simulate the diffusion process, a gradual transformation of data through a stochastic system. These models are particularly powerful for generating high-quality samples that follow a complex distribution.

### 🔹 Conditional Diffusion Models

**Conditional Diffusion Models** take the idea of diffusion models further by conditioning the generation process on auxiliary information, similar to cGANs. This enables the model to generate data that adheres to specific conditions or classes, making it a versatile tool for controlled data generation.

---

Explore the repository to dive deep into these models, experiment with the provided implementations, and witness the power of deep generative models in action!

---

## 📜 License

This project is licensed under the MIT License. See the [LICENSE](LICENSE) file for more details.