add empty check for input image

cospectrum · Jan 4, 2025 · 35b97bc · 35b97bc
1 parent d7c0aed
commit 35b97bc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/microwink/seg.py b/src/microwink/seg.py
@@ -83,6 +83,8 @@ def apply(
     ) -> list[SegResult]:
         CLASS_ID = 0
         assert image.mode == "RGB"
+        assert image.width > 0
+        assert image.height > 0
 
         raw = self._run(image, threshold.confidence, threshold.iou)
         if raw is None: