通過文件夾迭代圖像

Question

我有一個從0到9的灰度圖像的文件夾，它們大約是2400，我需要在python中加載它們以便將所有Zeors一起作為一個數組，然后將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數組中讀取完整數據的文件數。