Tweak augmentation

ankandrew · Apr 3, 2024 · e074f13 · e074f13
1 parent 506691e
commit e074f13
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/fast_plate_ocr/data/augmentation.py b/fast_plate_ocr/data/augmentation.py
@@ -10,9 +10,9 @@
 TRAIN_AUGMENTATION = A.Compose(
     [
         A.ShiftScaleRotate(
-            shift_limit=0.05,
+            shift_limit=0.06,
             scale_limit=0.1,
-            rotate_limit=8,
+            rotate_limit=9,
             border_mode=cv2.BORDER_CONSTANT,
             value=BORDER_COLOR_BLACK,
             p=1,
@@ -21,10 +21,10 @@
         A.MotionBlur(blur_limit=(3, 5), p=0.1),
         A.OneOf(
             [
-                A.CoarseDropout(max_holes=8, max_height=5, max_width=5, p=0.2),
+                A.CoarseDropout(max_holes=10, max_height=4, max_width=4, p=0.3),
                 A.PixelDropout(dropout_prob=0.01, p=0.2),
             ],
-            p=0.6,
+            p=0.7,
         ),
     ]
 )