fixed a bug in domain adaptation code

mingyuliutw · Oct 10, 2017 · 4cd5de5 · 4cd5de5
1 parent f42028d
commit 4cd5de5
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/datasets/dataset_mnist.py b/src/datasets/dataset_mnist.py
@@ -45,8 +45,7 @@ def _load_samples(self, full_filepath):
     # images = np.concatenate((images, images, images), axis=1)
     if self.use_inversion == 1:
       images = np.concatenate((images, 1 - images), axis=0)
-    images = 1 - images
-    labels = np.concatenate((labels, labels), axis=0)
+      labels = np.concatenate((labels, labels), axis=0)   
     images = (images - 0.5) * 2
     return np.float32(images), labels
 
@@ -102,4 +101,4 @@ def _load_samples(self, full_filepath):
     images = test_set[0]
     labels = test_set[1]
     images = (images - 0.5) * 2
-    return np.float32(images), labels
+    return np.float32(images), labels