Skip to content
Snippets Groups Projects
Commit ccdec8bf authored by JGlombitza's avatar JGlombitza
Browse files

Add Mnist Loader

parent b38b0779
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
Showing
with 2 additions and 2 deletions
MNIST_GAN/MNIST_GAN.py
+ 2
− 2
View file @ ccdec8bf
...@@ -2,7 +2,7 @@ import numpy as np ...@@ -2,7 +2,7 @@ import numpy as np
from keras.layers import Input from keras.layers import Input
from keras.models import Model from keras.models import Model
from keras.optimizers import Adam from keras.optimizers import Adam
import dlipr import mnist
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import keras.backend.tensorflow_backend as KTF import keras.backend.tensorflow_backend as KTF
from gan import generator_model, discriminator_model, plot_images, make_trainable, get_session from gan import generator_model, discriminator_model, plot_images, make_trainable, get_session
...@@ -11,7 +11,7 @@ from gan import generator_model, discriminator_model, plot_images, make_trainabl ...@@ -11,7 +11,7 @@ from gan import generator_model, discriminator_model, plot_images, make_trainabl
log_dir="." log_dir="."
KTF.set_session(get_session()) # Allows 2 jobs per GPU, Please do not change this during the tutorial KTF.set_session(get_session()) # Allows 2 jobs per GPU, Please do not change this during the tutorial
# prepare MNIST dataset # prepare MNIST dataset
data = dlipr.mnist.load_data() data = mnist.load_data()
X_train = data.train_images.reshape(-1, 28, 28, 1) / 255. X_train = data.train_images.reshape(-1, 28, 28, 1) / 255.
X_test = data.test_images.reshape(-1, 28, 28, 1) / 255. X_test = data.test_images.reshape(-1, 28, 28, 1) / 255.
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment