这是我用于 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,对于“3x16 位 rgb 颜色”的basn2c16.png,打印输出为:
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 缩放矩阵值相对容易,但是,您应该确保矩阵的格式正确...