从数据集中将图像加载到numpy矩阵数组中并应用转换

Question

我需要从一个数据集中将图像加载到一个numpy矩阵数组中，该数据集中包含800个图像，每个图像为64 x 64像素。我需要将每个64 x 64图像转换为具有4096列的矩阵行。下面，我显示了如何处理代码。我收到一个ValueError：无法将大小为4096的数组重塑为形状（64，）。请帮忙谢谢你。

array = np.zeros((800, 64))

for i in range(800):
    path = “some path”
    img = mpimg.imread(path)
    array[i] = img.reshape(64)

Answer 1

您的原始数组应为800, 4096形状，因为每个子数组代表一个(64, 64)图像，需要4096个元素。

因此，我认为您想要这个：

array = np.zeros((800, 4096))
paths = [...] # set paths here

for i, path in enumerate(paths):
    array[i] = mpimg.imread(path).reshape(4096)