将图像转换为值矩阵

Question

我的图像就像一块有4种颜色的棋盘（黑色，白色，红色，蓝色）。我必须将此图像转换为数字矩阵：1表示白色，2表示黑色，3表示红色等等。

例如图片：

应转换为矩阵：

[[1,2,1,2,1,2...]
[2,1,2,1,2,1...]
...]

我更喜欢python中的解决方案。

Answer 1

我不确定SVG图像，但我们假设你有一个PIL可读的图像格式（例如GIF，TIFF，JPEG，BMP，......）。然后你可以像这样用PIL读它：

import Image
img = Image.open("Chess_Board.bmp")

现在我们想要做quantization，所以图像像素不再是RGB，而是从0到3的颜色索引（假设你想要4种不同的颜色）：

quantized = img.convert('P', palette=Image.ADAPTIVE, colors=4)

接下来我想我们将它转​​换为numpy以便更容易地访问各个像素。然后我们做numpy魔术计算有多少进入一个区块：

import numpy as np
a = np.array(quantized)
blockLengthX = np.argmin(a[0]==a[0,0])
blockLengthY = np.argmin(a[:,0]==a[0,0])

之后很容易。我们只使用stepize blockLengthX为cols访问数组，而blockLengthY用于行：

result = a[::blockLengthX, ::blockLengthY]

当然，这假设您的所有块完全大小相同。 这是完整的程序，可以更轻松地复制和粘贴。我也缩短了一点：

import Image
import numpy as np
img = Image.open("Chess_Board.bmp")
a = np.array(img.convert('P', palette=Image.ADAPTIVE, colors=4))
blockLengthX = np.argmin(a[0]==a[0,0])
blockLengthY = np.argmin(a[:,0]==a[0,0])
result = a[::blockLengthX, ::blockLengthY]