如何在 numpy 数组中保存带有标签的多张图像？

Question

以前有人问过这类问题。但我的有点不同。例如，我在一个文件夹中有 20 个文件（20 个矩阵），每个矩阵的尺寸为 40*40。此外，这 20 个文件代表 20 个不同的类别。

现在我想创建一个 Numpy 数组，长度为 20*2。在每一行的第一列中，我想存储一个 40*40 的矩阵。因此，我要涵盖的所有 20 个文件。然后我想将标签保留为数字，例如1,2,3 .... 20. 稍后，当我调用 createdNumpyArray[0,0] 时，它应该显示第一个矩阵。如何在 Python 中执行此操作？

Answer 1

我会使用字典来存储从键到图像的映射。

ind_to_image = {
    0: numpy array with 40 x 40 shape, 
    ...,
    19: numpy array with 40 x 40 shape,
}

并将索引（键）保存到 20 x 2 数组的第一列因为我认为你不能直接将一个 numpy 数组保存到另一个数组的元素中。下面的代码会得到一个ValueError

arr1 = np.zeros((20, 2))
arr1[0, 0] = np.zeros((40, 40))

如何在 numpy 数组中保存带有标签的多张图像？

How can I save multiple images with labels in a numpy array?

python

numpy

multidimensional-array

numpy-ndarray