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

Python图像数据集处理：遍历并展示文件夹结构

原创已于 2023-04-18 22:45:01 修改 · 221 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #机器学习 #开发语言

于 2023-03-22 10:55:30 首次发布

该文描述了一个Python脚本，用于处理包含训练和验证数据集的dataset文件夹。它使用PIL库打开并显示图片，遍历train和val文件夹下的cat和dog子文件夹，收集图片路径和对应标签。

一、文件结构

本文的文件结构如下图所示，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()