aboutsummaryrefslogtreecommitdiff
path: root/dcgan.py
diff options
context:
space:
mode:
Diffstat (limited to 'dcgan.py')
-rw-r--r--dcgan.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/dcgan.py b/dcgan.py
index 61c0b48..1ffe50b 100644
--- a/dcgan.py
+++ b/dcgan.py
@@ -144,7 +144,7 @@ class DCGAN():
mu = np.mean(ref_imgs, axis=0)
sigma = np.var(ref_imgs, axis=0)
sigma[sigma<1] = 1
- img = np.divide(np.subtract(img, mu), sigma)
+ imgs = np.divide(np.subtract(imgs, mu), sigma)
# Sample noise and generate a batch of new images
noise = np.random.normal(0, 1, (batch_size, self.latent_dim))