python - PIL : fromarray gives a wrong object in P mode

Question

我想以 P 模式加载图像，将其转换为 np.array 然后再转换回来，但我得到了错误的 Image 对象，它是灰色的图片，不是彩色的

label = PIL.Image.open(dir).convert('P')
label = np.asarray(label)
img = PIL.Image.fromarray(label, mode='P')
img.save('test.png')

dir为原图片的路径； test.png是一张灰色图片

Answer 1

“P”模式下的图像需要一个将每个颜色索引与实际 RGB 颜色相关联的调色板。将图像转换为数组会丢失调色板，必须重新恢复它。

label = PIL.Image.open(dir).convert('P')
p = label.getpalette()
label = np.asarray(label)
img = PIL.Image.fromarray(label, mode='P')
img.setpalette(p)
img.save('test.png')