Skip to content

Instantly share code, notes, and snippets.

View dhanushkamath's full-sized avatar

Dhanush Kamath dhanushkamath

View GitHub Profile
import numpy as np
from keras.preprocessing.image import ImageDataGenerator
from keras.layers import Input, Conv2D, Flatten, Dense, Conv2DTranspose, Reshape, Lambda, Activation, BatchNormalization, LeakyReLU, Dropout
from keras.models import Model
from keras import backend as K
from keras.optimizers import Adam
from keras.callbacks import ModelCheckpoint
from keras.utils import plot_model
from google.colab import files
files.upload()
import os
from glob import glob
WEIGHTS_FOLDER = './weights/'
DATA_FOLDER = './data/img_align_celeba/'
if not os.path.exists(WEIGHTS_FOLDER):
os.makedirs(os.path.join(WEIGHTS_FOLDER,"AE"))
os.makedirs(os.path.join(WEIGHTS_FOLDER,"VAE"))