Correctly apply PNG palette when building ImageBase through Pillow.

anntzer · anntzer · commit ebe1fb22dc31 · 2019-05-25T23:50:38.000+02:00
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -667,6 +667,13 @@ def set_data(self, A):
         ----------
         A : array-like
         """
+        try:
+            from PIL import Image
+        except ImportError:
+            pass
+        else:
+            if isinstance(A, Image.Image):
+                A = pil_to_array(A)  # Needed e.g. to apply png palette.
         self._A = cbook.safe_masked_invalid(A, copy=True)
 
         if (self._A.dtype != np.uint8 and
