Python中合并R、G和B分量时颜色不匹配

Question

在下面的代码中，我首先将图像转换为Numpy数组：

import numpy as np
from PIL import Image

sceneImage = Image.open('peppers.tif')
arrImage = np.array(sceneImage) # 256x256x3 array

现在我定义了一个新数组，并按以下代码将arrImage放入其中：

import matplotlib.pyplot as plt
final_image = np.zeros((N,N,3))
final_image[...,0] = arrImage[...,0]
final_image[...,1] = arrImage[...,1]
final_image[...,2] = arrImage[...,2]
plt.imshow(final_image)
plt.show()

换句话说，我已经单独复制粘贴了R、G和B分量（尽管这可以更轻松地完成，但我是故意这样做的）。

现在的问题是fianl_image和arrImage对我来说显示的图像不同。形状看起来像下面这样：

arrImage: final_image: 这两个之间的问题是什么？

- A.M.

---

你是如何显示这两个数组的？看起来你已经将一个YUV图像显示为RGB了。 - Daniel

这里的 N 是什么？ - Divakar

final_image = np.zeros((N, N, 3),dtype=np.uint8) - Padraic Cunningham

@Divakar 这是图片的尺寸大小。 - A.M.

有趣的是，这两张图片完美地互为反面。 - dpwilson

@PadraicCunningham 谢谢。那解决了问题。您能否将其添加到答案部分，以便我可以接受它？ - A.M.

Answer 1

从文档中： dtype : 数据类型，可选 所需的数组数据类型，例如numpy.int8。默认值为numpy.float64。

你只需要将正确的dtype传递给np.zeros，即np.uint8：

 sceneImage = Image.open('peppers.tif')
arrImage = np.array(sceneImage)  # 256x256x3 array
final_image = np.zeros((N, N,  3),dtype=np.uint8)
final_image[..., 0] = arrImage[..., 0]
final_image[..., 1] = arrImage[..., 1]
final_image[..., 2] = arrImage[..., 2]

img = Image.fromarray(final_image, 'RGB')
img.save('whatever.png')

- Padraic Cunningham