Merge pull request #432 from kwcckw/dev

Fixed bug in NoiseTexturize.
sparkfish · Apr 2, 2024 · 4132722 · 4132722
2 parents 2a36dba + f78eab4
commit 4132722
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/augraphy/augmentations/noisetexturize.py b/augraphy/augmentations/noisetexturize.py
@@ -94,9 +94,14 @@ def __call__(self, image, layer=None, mask=None, keypoints=None, bounding_boxes=
                 image, image_alpha = image[:, :, :3], image[:, :, 3]
 
             sigma = random.randint(self.sigma_range[0], self.sigma_range[1])
-            turbulence = random.randint(
-                self.turbulence_range[0],
-                self.turbulence_range[1],
+
+            # turbulence must > 1 to prevent endless while loop below
+            turbulence = max(
+                2,
+                random.randint(
+                    self.turbulence_range[0],
+                    self.turbulence_range[1],
+                ),
             )
 
             result = image.astype(float)