python - 通过文件夹迭代图像

Question

我有一个从 0 到 9 的灰度图像文件夹，它们大约是 2400，我需要将它们加载到 python 中，以便将所有 Zeor 作为一个数组放在一起，然后将 Ones 作为一个数组放在一起，等等......我用过下面的代码将一个图像加载为一个数组，但我不知道如何加载所有图像并将每个数字组合在一起。我考虑过通过文件夹进行迭代。

有谁知道该怎么做或有其他想法吗？

import imageio

im = imageio.imread('Train/1.jpg')

Answer 1

您可以通过以下方式执行此操作：

import imageio
for i in range(9):
    im = imageio.imread('Train/'+str(i)+'.jpg')

您还可以创建一个 3D 变量，其第三维等于文件数，以读取单个 3D 数组中的完整数据。