Bugfix

asanakoy · Feb 21, 2018 · 5823438 · 5823438
1 parent d3a7413
commit 5823438
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/albu/src/transforms.py b/albu/src/transforms.py
@@ -109,7 +109,7 @@ def __call__(self, img, mask=None):
             factor = random.randint(0, 4)
             img = np.rot90(img, factor)
             mask = np.rot90(mask, factor)            
-            return img, mask
+        return img, mask
 
 
 class Rotate: