Merge pull request #450 from colibrisson/fix_augment

disable opencv multithreading
mittagessen · Feb 21, 2023 · 48c7433 · 48c7433
2 parents d81e898 + 8e297b4
commit 48c7433
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/kraken/lib/dataset/recognition.py b/kraken/lib/dataset/recognition.py
@@ -101,6 +101,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,

diff --git a/kraken/lib/dataset/segmentation.py b/kraken/lib/dataset/segmentation.py
@@ -157,6 +157,8 @@ def __init__(self, imgs: Sequence[Union[PathLike, str]] = None,
         else:
             raise Exception('invalid dataset mode')
         if augmentation:
+            import cv2
+            cv2.setNumThreads(0)
             from albumentations import (
                 Compose, ToFloat, RandomRotate90, Flip, OneOf, MotionBlur, MedianBlur, Blur,
                 ShiftScaleRotate, OpticalDistortion, ElasticTransform,