Merge pull request #423 from Aikhjarto/fix_422

don't multiply by 255 is data is already uint8
nschloe · Jul 24, 2020 · da6335a · da6335a
2 parents 32a3bb9 + 4972f62
commit da6335a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tikzplotlib/_image.py b/tikzplotlib/_image.py
@@ -34,7 +34,9 @@ def draw_image(data, obj):
             img_array = numpy.flipud(img_array)
 
         # Convert mpl image to PIL
-        image = PIL.Image.fromarray(numpy.uint8(img_array * 255))
+        if img_array.dtype != numpy.uint8:
+            img_array = numpy.uint8(img_array * 255)
+        image = PIL.Image.fromarray(img_array)
 
         # If the input image is PIL:
         # image = PIL.Image.fromarray(img_array)