在我们做深度学习的数据处理的时候,需要用到数据集的 train.txt ,也就是遍历指定路径文件夹下面的图片路径和图片名字,然后生成一个 train.txt ,下面是python的一种遍历方法。
import os
import numpy as np
def create_txt(name, path, file_image):
txt_path = path + name + '.txt'
txt = open(txt_path, 'w')
for i in (file_image):
print(i)
image_dir = os.path.join('C:/Users/lenovo/Desktop/打标分类/sum/sample/',str(i)+'\n')
txt.write(image_dir)
def read_file(path1):
filelist1 = os.listdir(path1)
file_image = [file for file in filelist1 if file.endswith('.jpg')]
return file_image
path1 = 'C:/Users/lenovo/Desktop/打标分类/sum/sample/'
file_image = read_file(path1)
create_txt('train', '', file_image)
运行后生成的 train.txt 会在这个 py 文件的同级目录下。