aboutsummaryrefslogtreecommitdiff
path: root/cgan.py
diff options
context:
space:
mode:
authornunzip <np.scarh@gmail.com>2019-02-27 18:50:58 +0000
committernunzip <np.scarh@gmail.com>2019-02-27 18:50:58 +0000
commit8268b98b17dc5190eb5c12c914ff4ffcc504400b (patch)
tree522042995075c97ec9c4f006da8051443a340b80 /cgan.py
parent7eec33f756cffee0c4526c7e3e6af71246dd0787 (diff)
downloade4-gan-8268b98b17dc5190eb5c12c914ff4ffcc504400b.tar.gz
e4-gan-8268b98b17dc5190eb5c12c914ff4ffcc504400b.tar.bz2
e4-gan-8268b98b17dc5190eb5c12c914ff4ffcc504400b.zip
Remove padding
Diffstat (limited to 'cgan.py')
-rw-r--r--cgan.py5
1 files changed, 0 insertions, 5 deletions
diff --git a/cgan.py b/cgan.py
index d44f961..7bef77c 100644
--- a/cgan.py
+++ b/cgan.py
@@ -203,11 +203,6 @@ class CGAN():
labels_test[i*100:] = i
train_data = self.generator.predict([noise_train, labels_train])
test_data = self.generator.predict([noise_test, labels_test])
- test_data = np.pad(test_data, ((0,0),(2,2),(2,2),(0,0)), 'constant')
- train_data = np.pad(train_data, ((0,0),(2,2),(2,2),(0,0)), 'constant')
- labels_test = labels_test.flatten()
- labels_train = labels_train.flatten()
- print(train_data.shape, test_data.shape, labels_test.shape, labes_train.shape)
return train_data, test_data, labels_train, labels_test