如何使用Pillow获取浮动RGBA像素值列表?

时间:2016-11-25 18:32:11

标签: python pillow

我想使用Pillow python模块获取浮动RGBA像素值的列表。

到目前为止,我只能得到RGBA整数数据:

from PIL import Image

im = Image.open("Lenna.png")
im_alpha = im.convert('RGBA')
Pixels = list(im.getdata())

这会让我举例((226,137,125,255),...) 但是,我不知道如何以浮点形式获取此信息,例如((0.88627451,0.537254902,0.490196078,1),...)。

我该怎么做?

1 个答案:

答案 0 :(得分:1)

我正在使用RGB .png:

from PIL import Image
import numpy

# http://www.schaik.com/pngsuite/basn2c16.png
im = Image.open('basn2c16.png')
#~ data = numpy.asarray(im)
data = numpy.array(im) # same as .asarray
print("Array dimensions: %s"%(repr(data.shape)))
data = data.astype(float)
print("[20, 30]=%s"%(repr(data[20, 30])))
print(data)
data = numpy.divide(data, 255.0)
print(data)

现在,请注意它取决于.png的类型;例如,请参阅http://www.schaik.com/pngsuite/pngsuite_bas_png.html了解测试文件,而basn2c16.png是“3x16位rgb颜色”,打印输出为:

Array dimensions: (32, 32, 3)
[20, 30]=array([   8.,   90.,  156.])
[[[ 255.  255.    0.]
  [ 247.  255.    0.]
  [ 239.  255.    0.]
...
[[[ 1.          1.          0.        ]
  [ 0.96862745  1.          0.        ]
  [ 0.9372549   1.          0.        ]
  ..., 

因此,即使这是16位,值似乎跨越0-255,就像8位一样。在这种情况下,我们需要使用numpy.divide和255来缩放数据,以获得0.0-1.0的浮动范围......

但是,如果你使用索引/托管png basn3p08.png,你会得到:

Array dimensions: (32, 32)
[20, 30]=120.0
[[ 165.  165.  165. ...,  254.  254.  254.]
 [   8.    8.    8. ...,  248.  248.  248.]
....

所以现在矩阵内容不代表RGB(A)值,而是代表调色板中的索引,所以除以255就没有意义。

最后,如果它是像basn6a16.png这样的RGBA png,你也会获得alpha:

Array dimensions: (32, 32, 4)
[20, 30]=array([   0.,   88.,  167.,   16.])
[[[ 255.  255.    0.    0.]
  [ 247.  255.    0.    0.]
...

同样,即使这是16位png,值似乎也缩放为255,因此除以255以获得浮点数是有道理的。

换句话说,使用numpy和PIL比较容易缩放矩阵值,但是,您应该确保矩阵的格式正确...