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检查它是否是图片结尾的文件。