python生成数据集txt文件与读取数据集

目录

 

一、根据文件夹自动生成带label的txt文件：

二、根据txt文件自动生生成filename，label列表数据：

---

 

一、根据文件夹自动生成带label的txt文件：

这是用于给图片打标签并生成txt文件的程序；

creat_labeled_txt  给三级目录里的文件打标签，并且标签根据第二级级目录自动生成，如shan:/right(wrong):/a.jpg

 

在下面程序里主要用了这几个功能：

1、os.listdir('dir')：用于看某一目录下的文件名，返回list

2、txt文件写入：

            f = open('dir', mode)

           mode = 'w' 时，是文件覆盖写入；

           mode = 'a'时，文件是在最后添加写入；

           f.write(str + '\n')

 

程序：

import os
#分三级目录，如A/B/a.jpg
#input_path为一级目录；
#
def creat_filelist(input_path, classes):
#创建三级目录
#index 一定是str类型，不可以为int
    dir_image1 = [] #二级目录
    file_list = [] #三级目录
    for index, name in enumerate(classes):
        print('index', index)
        index_str = str(index)
        dir_image1_temp = input_path + '/' + name &#