如何從python中的不同文件夾和子文件夾加載圖像

Question

我正在動物分類數據集中開發CNN，該數據被分成2個文件夾，並且這2個文件夾包含另一個子文件夾...此結構有四層，現在我想加載它們並將其轉換為n維-arrays饋送到tensorflow，這些文件夾的名稱是標簽。 我希望有人可以為我提供一些具體的代碼或有用的材料。 提前非常感謝您！

在這里，我將舉一些例子：Anisopleura Libellulidae Leach，1815年Trithemis aurora Zygoptera Calopterygidae Selys，1850 Calopteryx錦綉此問題的標簽，它們也是五樓子文件夾的名稱，圖像存儲在這些文件夾中。 C：\\ Users \\ Seth \\ Desktop \\ dragonfly \\ Anisopleura \\ Libellulidae Leach，1815 \\ Pseudothemis \\ zonata

這是一條路。

Answer 1

我使用openface庫進行人臉識別，在該庫中，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多維數組）以為每個標簽（或子文件夾）存儲圖像的特征。