Merge pull request #35 from albu/cutout

Default probability for Cutout = 0.5
albumentations-team · Jul 24, 2018 · 0772e72 · 0772e72
2 parents e01d84a + a4f9aa4
commit 0772e72
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/albumentations/augmentations/transforms.py b/albumentations/augmentations/transforms.py
@@ -410,7 +410,7 @@ class Cutout(ImageOnlyTransform):
 
         """
 
-    def __init__(self, num_holes=0, max_h_size=0, max_w_size=0, p=1.0):
+    def __init__(self, num_holes=0, max_h_size=0, max_w_size=0, p=0.5):
         super(Cutout, self).__init__(p)
         self.num_holes = num_holes
         self.max_h_size = max_h_size