Added a min size of 1 to avoid zero width/length in resize (#250)

cleanlab · Feb 1, 2024 · 4b4932d · 4b4932d
1 parent 53ee321
commit 4b4932d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cleanvision/issue_managers/image_property.py b/src/cleanvision/issue_managers/image_property.py
@@ -254,7 +254,7 @@ def calculate(self, image: Image) -> Dict[str, Union[float, str]]:
         ratio = max(image.width, image.height) / self.max_resolution
         if ratio > 1:
             resized_image = image.resize(
-                (int(image.width // ratio), int(image.height // ratio))
+                (max(int(image.width // ratio), 1), max(int(image.height // ratio), 1))
             )
         else:
             resized_image = image.copy()