在Python中组合R,G和B组件时的颜色不匹配

2024年1月3日 197次阅读

在下面的代码中,我首先将图像转换为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：
《在Python中组合R,G和B组件时的颜色不匹配》

final_image：
《在Python中组合R,G和B组件时的颜色不匹配》

这两个不一样的问题是什么？

最佳答案 从
docs

dtype：数据类型,可选

The desired data-type for the array, e.g., numpy.int8. Default is 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')
点赞