Merge pull request #5879 from radarhere/quantize

Limit quantized palette to number of colors
python-pillow · Dec 28, 2021 · 17ec8b5 · 17ec8b5
2 parents 39de053 + ec19889
commit 17ec8b5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/Tests/test_image_quantize.py b/Tests/test_image_quantize.py
@@ -77,6 +77,13 @@ def test_quantize_dither_diff():
     assert dither.tobytes() != nodither.tobytes()
 
 
+def test_colors():
+    im = hopper()
+    colors = 2
+    converted = im.quantize(colors)
+    assert len(converted.palette.palette) == colors * len("RGB")
+
+
 def test_transparent_colors_equal():
     im = Image.new("RGBA", (1, 2), (0, 0, 0, 0))
     px = im.load()

diff --git a/src/PIL/Image.py b/src/PIL/Image.py
@@ -1109,7 +1109,8 @@ def quantize(self, colors=256, method=None, kmeans=0, palette=None, dither=1):
         from . import ImagePalette
 
         mode = im.im.getpalettemode()
-        im.palette = ImagePalette.ImagePalette(mode, im.im.getpalette(mode, mode))
+        palette = im.im.getpalette(mode, mode)[: colors * len(mode)]
+        im.palette = ImagePalette.ImagePalette(mode, palette)
 
         return im