python - 将图像 ( png ) 转换为矩阵，然后转换为一维数组

Question

我有 5 张图片，我想将每个图像转换为一维数组并将其作为矢量放入矩阵中。
我希望能够再次将每个矢量转换为图像。

img = Image.open('orig.png').convert('RGBA')
a = np.array(img)

我不熟悉 numpy 的所有功能，想知道是否还有其他工具可以使用。
谢谢。

Answer 1

import numpy as np
from PIL import Image

img = Image.open('orig.png').convert('RGBA')
arr = np.array(img)

# record the original shape
shape = arr.shape

# make a 1-dimensional view of arr
flat_arr = arr.ravel()

# convert it to a matrix
vector = np.matrix(flat_arr)

# do something to the vector
vector[:,::10] = 128

# reform a numpy array of the original shape
arr2 = np.asarray(vector).reshape(shape)

# make a PIL image
img2 = Image.fromarray(arr2, 'RGBA')
img2.show()