A bunch of GAN implementations
Project description
GANs-Implementations
GANs Implementations and other generative models + Training (in ./notebooks)
Implemented:
- Vanilla GAN
- DCGAN - Deep Convolutional GAN
- WGAN - Wasserstein GAN
- SNGAN - Spectrally Normalized GAN
- SRGAN - Super Resolution GAN
- StyleGAN
- Pix2PixHD
- C-VAE - Convolutional Variational Auto-encoder
Installation
$ pip install gans-implementations
Local Install and Run:
$ cd {PROJECT_DIRECTORY}
$ pip install -e .
Example
In notebooks directory there is a notebook on how to use each of these models for their intented use case; such as image generation for StyleGAN and others. Check them out!
from gans_package.models import StyleGAN_Generator, StyleGAN_Discriminator
in_channels = 256
out_channels = 3
hidden_channels = 512
z_dim = 128
mapping_hidden_size = 256
w_dim = 512
synthesis_layers = 5
kernel_size=3
in_size = 3
d_hidden_size = 16
g = StyleGAN_Generator(in_channels,
out_channels,
hidden_channels,
z_dim,
mapping_hidden_size,
w_dim,
synthesis_layers,
kernel_size,
device=DEVICE).to(DEVICE)
d = StyleGAN_Discriminator(in_size, d_hidden_size).to(DEVICE)
import torch
noise = torch.randn(BATCH_SIZE, z_dim).to(DEVICE)
fake = g(noise)
pred = d(fake)
Handwritten Digits - MNIST
Work Cited
https://arxiv.org/pdf/1609.04802v5.pdf
https://arxiv.org/pdf/1812.04948.pdf
https://www.coursera.org/specializations/generative-adversarial-networks-gans?
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for gans_implementations-0.1.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 485f155be3f3edf0a60b4d8856f2a48c4825b1a1cb2194e67ac2078dd04b4306 |
|
MD5 | 33651806793f510dc596964f22cd2f68 |
|
BLAKE2b-256 | e8fc516b095f0caab3fc7aa5ca6608aaef713f1497a20289735c75e0ac9a8fbe |
Close
Hashes for gans_implementations-0.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e62156a4b82e8137c6c86f1ac24aa7897fceeffac99f7afe50dfc0366cf95e23 |
|
MD5 | 95a8cab41bf2ecb686de1fa87b4c7e64 |
|
BLAKE2b-256 | 8b4c7773efc97eacabcb046ebcf0b8c7aa259a46359455495e1e0b516b890d6e |