python - Python tflearn：加载包含图像文件夹的 zip 文件

Question

我有一个 zip 文件，其中有数百个文件夹，每个文件夹包含 3-4 张图像。我想用这些图像作为训练数据来识别人脸。如何加载这些图像？而且，加载数据集的最佳方式是什么？作为 pkl 文件还是其他文件？

Answer 1

我会先解压缩文件，然后你有一个文件夹结构，然后你可以轻松地做这样的事情：

import os
imgs = []
for directory in os.listdir(base_directory):
    for file in os.listdir(directory):
         imgs.append(scipy.misc.imread(base_directory + "/" + directory + "/" + file)

如果您有其他文件，您可以使用os.path.isdir它来检查它是否是一个目录并file.endswith检查它是否是一个以您的图像结尾的文件。