aug

bmd1905 · Nov 29, 2020 · d424594 · d424594
1 parent b00660d
commit d424594
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vietocr/loader/aug.py b/vietocr/loader/aug.py
@@ -16,7 +16,7 @@ def __init__(self):
 
         # color
         sometimes(iaa.AddToHueAndSaturation(value=(-10, 10), per_channel=True)),
-        sometimes(iaa.SigmoidContrast(gain=(3, 10), cutoff=(0.4, 0.6)), per_channel=True),
+        sometimes(iaa.SigmoidContrast(gain=(3, 10), cutoff=(0.4, 0.6), per_channel=True)),
         sometimes(iaa.Invert(0.25, per_channel=0.5)),
         sometimes(iaa.Solarize(0.5, threshold=(32, 128))),
         sometimes(iaa.Dropout2d(p=0.5)),