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