Lower rotation in line strip dataset augmentation

Might be related to #482
mittagessen · Apr 12, 2023 · dd846ef · dd846ef
1 parent c36256c
commit dd846ef
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/kraken/lib/dataset/recognition.py b/kraken/lib/dataset/recognition.py
@@ -305,6 +305,8 @@ def __init__(self,
             else:
                 self.text_transforms.append(F_t.text_reorder)
         if augmentation:
+            import cv2
+            cv2.setNumThreads(0)
             from albumentations import (
                 Compose, ToFloat, OneOf, MotionBlur, MedianBlur, Blur,
                 ShiftScaleRotate, OpticalDistortion, ElasticTransform,
@@ -496,6 +498,8 @@ def __init__(self, split: Callable[[Union[PathLike, str]], str] = F_t.default_sp
             else:
                 self.text_transforms.append(F_t.text_reorder)
         if augmentation:
+            import cv2
+            cv2.setNumThreads(0)
             from albumentations import (
                 Compose, ToFloat, OneOf, MotionBlur, MedianBlur, Blur,
                 ShiftScaleRotate, OpticalDistortion, ElasticTransform,
@@ -508,7 +512,7 @@ def __init__(self, split: Callable[[Union[PathLike, str]], str] = F_t.default_sp
                                     MedianBlur(blur_limit=3, p=0.1),
                                     Blur(blur_limit=3, p=0.1),
                                 ], p=0.2),
-                                ShiftScaleRotate(shift_limit=0.0625, scale_limit=0.2, rotate_limit=45, p=0.2),
+                                ShiftScaleRotate(shift_limit=0.0625, scale_limit=0.2, rotate_limit=3, p=0.2),
                                 OneOf([
                                     OpticalDistortion(p=0.3),
                                     ElasticTransform(p=0.1),