python - 使用 pillow/numpy 在任何拆分带上执行 .dot 后无法合并拆分图像

Question

我正在尝试使用 pillow 和 numpy 校正图像中的颜色。将 im.split() 与 np.array 结合使用。

我想乘以红色带中的所有颜色，但找不到实现它的方法。

我尝试了各种方法，经过大量谷歌搜索后，我希望这将是解决方案:

from PIL import Image
import numpy as np

im=Image.open('test.jpg')
r,g,b=im.split() 

datar = np.array(r)
datag = np.array(g)
datab = np.array(b)

rm=0.4 # the value I would like to multiply all red pixels by

datar=datar.dot(rm) # this works, but turns the values in the array into floats
datar=datar.astype(int) # I was hoping this would solve it

im=Image.merge("RGB", (Image.fromarray(datar), Image.fromarray(datag), Image.fromarray(datab)))

我可以用数组做很多事情并且合并成功，但是尝试这样做会给我以下错误:

im=Image.merge("RGB", (Image.fromarray(datar), Image.fromarray(datag), Image.fromarray(datab)))
  File "/usr/local/lib/python2.7/dist-packages/PIL/Image.py", line 2408, in merge
im.putband(bands[i].im, i)
ValueError: images do not match

数组在应用 .dot 和 .astype(int) 之前和之后看起来是一样的，并且值正确相乘。

Answer 1

Image.merge 失败，因为 rgb 图像的模式不同(参见 PIL Image modes )。您可以通过这种方式检查模式:

>>> Image.fromarray(datar).mode
'I'
>>> Image.fromarray(datag).mode
'L'

原因是numpy数组的类型:

>>> datar.dtype
dtype('int32')
>>> datag.dtype
dtype('uint8')

要解决这个问题，请替换为:

datar=datar.astype(int)

用这个:

datar = datar.astype('uint8')