如何从 Google Colab 的“train”文件夹中的多个文件夹加载多个图像？

Question

我正在将 Google Colab 用于食品分类的 ML 模型。 我的火车文件夹包含 6 个子文件夹。 我想读取所有图像并将其分配给 train_images 变量，如图所示。 问题是我有 6 个文件夹，每个文件夹中有 40 张图片（.jpg）。 那么代码应该是什么呢？ 我为此使用python。

请帮忙！

Answer 1

这可能有帮助：

>>> for folder in os.listdir('desktop/images'):
...     for image in os.listdir(f'desktop/images/{folder}'): print(image)
...
20160327_174551.jpg
20160327_174559.jpg
20160327_174636.jpg
20160327_174718.jpg
20160327_174756.jpg
20160327_174810.jpg

图像目录中有六个文件夹，每个文件夹中有一个图像。 这两个 for 循环将找到所有图像。