小编典典

如何将 NumPy 数组转换为应用 matplotlib 颜色图的 PIL 图像

all

我有一个简单的问题,但我找不到一个好的解决方案。

我想采用代表灰度图像的 NumPy 2D 数组,并在应用一些 matplotlib 颜色图时将其转换为 RGB PIL 图像。

pyplot.figure.figimage我可以使用以下命令获得合理的 PNG 输出:

dpi = 100.0
w, h = myarray.shape[1]/dpi, myarray.shape[0]/dpi
fig = plt.figure(figsize=(w,h), dpi=dpi)
fig.figimage(sub, cmap=cm.gist_earth)
plt.savefig('out.png')

虽然我可以调整它以获得我想要的东西(可能使用 StringIO 得到 PIL
图像),但我想知道是否没有更简单的方法可以做到这一点,因为它似乎是图像可视化的一个非常自然的问题。比方说,像这样:

colored_PIL_image = magic_function(array, cmap)

阅读 76

收藏
2022-06-06

共1个答案

小编典典

相当繁忙的单线,但这里是:

  1. 首先确保您的 NumPy 数组 ,myarray用 处的最大值进行归一化1.0
  2. 将颜色图直接应用于myarray.
  3. 重新调整0-255范围。
  4. 转换为整数,使用np.uint8().
  5. 使用Image.fromarray().

你完成了:

from PIL import Image
from matplotlib import cm
im = Image.fromarray(np.uint8(cm.gist_earth(myarray)*255))

plt.savefig()

在此处输入图像描述

im.save()

在此处输入图像描述

2022-06-06