fix the flip bug

seanXYZ · Jun 25, 2019 · e2e4b07 · e2e4b07
1 parent 6215075
commit e2e4b07
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dataset.py b/dataset.py
@@ -50,9 +50,9 @@ def __getitem__(self, index):
                 mask = VF.hflip(mask)
                 img  = VF.hflip(img)
 
-            # if np.random.rand(1) > 0.5:
-            #     mask = VF.vflip(mask)
-            #     img  = VF.vflip(img)
+            if np.random.rand(1) > 0.5:
+                mask = VF.vflip(mask)
+                img  = VF.vflip(img)
 
             img = self.transform(img)
             mask = self.maskTransform(mask)