aboutsummaryrefslogtreecommitdiff
path: root/cgan.py
diff options
context:
space:
mode:
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