os.walk()方法读取图像集标签与路径到列表

最新推荐文章于 2022-06-10 21:54:39 发布

原创最新推荐文章于 2022-06-10 21:54:39 发布 · 428 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch

数据处理专栏收录该内容

4 篇文章

订阅专栏

该代码段遍历'train'文件夹下的所有图像文件，将每个图像的完整路径添加到data_path列表中，并将不带扩展名的图像文件名添加到data_name列表中，为后续的图像处理或机器学习任务做准备。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os

data_path = []
data_name = []

for root, dirs, files in os.walk('train'):
    # 变量指定目录文件列表
    for image_file in files:
        image_path = os.path.join(root, image_file)
        data_path.append(image_path)
        data_name.append(image_file.split('.')[0])

train文件夹下为图像数据集：