diff options
| -rw-r--r-- | cgan.py | 5 |
1 files changed, 0 insertions, 5 deletions
@@ -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 |