Merge pull request #13 from SagerKudrick/master

Update palette.py
hbldh · Sep 18, 2023 · 0f3bbc4 · 0f3bbc4
2 parents fa481f9 + 6f48556
commit 0f3bbc4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/hitherdither/ordered/bayer.py b/hitherdither/ordered/bayer.py
@@ -42,6 +42,8 @@ def I(n, transposed=False):
     :return: The index matrix.
 
     """
+    if n == 0:
+        return np.array([[0, 0], [0, 0]], "int")
     if n == 2:
         if transposed:
             return np.array([[0, 3], [2, 1]], "int")

diff --git a/hitherdither/palette.py b/hitherdither/palette.py
@@ -182,7 +182,8 @@ def median_cut(p, dim=None):
         while len(bins) < n:
             new_bins = []
             for bin in bins:
-                new_bins += median_cut(bin, dim)
+                if len(bin) != 0:
+                    new_bins += median_cut(bin, dim)
             bins = new_bins
 
         # Average over pixels in each bin to create