获取文件夹下所有文件的路径，并以文件夹名称为标签

一、文件结构

  本文的文件结构如下图所示，dataset文件夹下有两个文件夹，分别为train和val，其下又分别有cat和dog两个文件夹。

二、程序

from PIL import Image
import os
root_dir = 'dataset/train'
filenames = []
labels = []
all_classes = os.listdir(root_dir)
for single_class in all_classes:
    single_class_path = os.path.join(root_dir, single_class)
    all_images = os.listdir(single_class_path)
    for single_image in all_images:
        filenames.append(os.path.join(single_class_path, single_image))
        labels.append(single_class)     

img = Image.open(filenames[0])
img.show()