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来检查它是否是一个以你的图像结尾的文件。