A Deep Convolutional GAN created for generation of low-medium (64x64 px) resolution images.
Demonstration can be found in https://konkuad.github.io/gan.html, which is created and run on a google colab notebook.
The dataset used in the demo can be downloaded from https://www.kaggle.com/prasoonkottarathil/gananime-lite.
A 64x64 processed version (used in my google drive) can be found here https://drive.google.com/file/d/1zUOt42VfZ9jaPNtwZhZOV4pf2mCx0uWL/view?usp=sharing.
To install, in terminal
git clone https://github.com/konkuad/GANime
cd GANime
pip install .
To create a dataloader from an image folder
from torch.utils.data import DataLoader
import torchvision.transforms as T
from GANime.gan import plotter
from GANime.datasets import ImageOnlyDataset
resize_transform = T.Compose([
T.Resize(64), #resize
T.ToTensor(), #convert to tensor
T.Lambda(lambda x: (255*x).int()/127.5-1) #normalize color channels to -1 and 1
])
ds = ImageOnlyDataset('out2', resize_transform)
dl = DataLoader(ds, batch_size=128, shuffle=True)
#plot a few images
it = next(iter(dl))
plotter(it, rows=8, columns=8, renormalize_func = lambda x: (x*127.5+127.5).astype(int))Example of using the package on your dataloader.
from GANime.gan import GAN
seed_size = 128
gan_model = GAN(seed_size)
gan_model.train(dl,
num_epochs = 20,
batch_size = 128,
plot = True,)