通过文件夹迭代图像

Question

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

有人知道怎么做吗？或者还有其他想法吗？

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阵列中读取完整数据的文件数。