#11

ildoonet · Jan 29, 2020 · 48b8f50 · 48b8f50
1 parent 09c0799
commit 48b8f50
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/RandAugment/augmentations.py b/RandAugment/augmentations.py
@@ -90,7 +90,8 @@ def SolarizeAdd(img, addition=0, threshold=128):
 
 
 def Posterize(img, v):  # [4, 8]
-    v = int(v) + 1
+    v = int(v)
+    v = max(1, v)
     return PIL.ImageOps.posterize(img, v)