python遍历文件夹生成标签数据txt文件

最新推荐文章于 2024-04-23 14:07:17 发布

原创最新推荐文章于 2024-04-23 14:07:17 发布 · 1.4k 阅读

8 ·

CC 4.0 BY-SA版权

python 同时被 2 个专栏收录

25 篇文章

订阅专栏

caffe

14 篇文章

订阅专栏

本文介绍了一种使用Python进行目录遍历并记录文件路径的方法。通过os模块获取指定路径下的所有子目录及图片文件，并将这些文件路径及对应类别编号写入到一个txt文件中。此方法适用于为机器学习项目准备数据集。

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

# coding=UTF-8

import os
import sys

path="/home/cdli/Original_10.18"
filenames=os.listdir(path) #读取path内所有文件名返回列表
i=0
with open("train_color.txt","w") as f:
        for filename in filenames:
                sub_path=path+"/"+filename+"/"
                pic=os.listdir(sub_path)
                print(filename+','+str(i))
                for p in pic:
                        f.write('XXX/'+filename+'/'+p+' '+str(i)+'\n')
                i=i+1