如何在python中从不同的文件夹和子文件夹加载图像

Question

我正在开发一个动物分类数据集中的CNN，分为2个文件夹，2个文件夹涉及到另外一个子文件夹……这个结构有4层，现在我想加载它们并将它们转换为n-dimension-arrays 提供给 tensorflow，这些文件夹的名称是标签。 我希望有人可以帮助我提供一些具体的代码或一些有用的材料。 非常感谢您！

这里我举几个例子： Anisopleura Libellulidae Leach, 1815 Trithemis aurora Zygoptera Calopterygidae Selys, 1850 Calopteryx splendens 极光和芨芨草是这个问题的标签，也是五楼子文件夹的名称，图片存放在这些文件夹中。 C:\Users\Seth\Desktop\dragonfly\Anisopleura\Libellulidae Leach, 1815\Pseudothemis\zonata

这是一条路径。

Answer 1

我使用 openface 库进行人脸识别，在这个库中 iterImgs 是一种方法，可以列出目录下的所有图像

详情iterImgs

from openface.data import iterImgs

imgs = list(iterImgs("Directory path"))

print imgs    # print all images in Directory path also in Tree

或者另一种方式是定义一个有效的扩展

vailed_ext = [".jpg",".png"]
import os 
f_list = []
def Test2(rootDir):     
    for lists in os.listdir(rootDir): 
        path = os.path.join(rootDir, lists) 
        filename, file_extension = os.path.splitext(path) 
        if file_extension in vailed_ext:
            print path          
            f_list.append[path]
        if os.path.isdir(path): 
           Test2(path)

Test2("/home/")
print f_list

Answer 2

os.walk() 就是你要找的东西。

import os

# traverse root directory, and list directories as dirs and files as files
for root, dirs, files in os.walk("."):
    path = root.split(os.sep)
    print((len(path) - 1) * '---', os.path.basename(root))
    for file in files:
        print(len(path) * '---', file)

此代码将允许您递归解析所有文件夹和子文件夹。您将获得子文件夹的名称（在您的情况下为标签）和 file 变量中的所有文件。

接下来的工作可能是创建一个字典（或 numpy 多维数组）来为每个标签（或子文件夹）存储图像的特征。