import os dir=r'C:\Users\huawei\Desktop\pythonProject1\data\train' fp = open('./train_list.txt','w+') imgfile_list = os.listdir(r'C:\Users\huawei\Desktop\pythonProject1\data\train') imgfile_list.sort(key= lambda x:int(x[:])) #print(img_list) seqsize =17 for imgfile in imgfile_list: filepath = os.path.join(dir,imgfile) img_list = os.listdir(filepath) img_list.sort(key=lambda x: int(x[:-4])) #滑窗取序列,步长为8 for i in range(0, len(img_list)-seqsize, 8): for j in range(i,i+seqsize): img = img_list[j] path = os.path.join(filepath, img) if j == i+seqsize-1: fp.write(path+'\n') else: fp.write(path+' ') fp.close()上述代码去掉排序部分
时间: 2023-06-09 22:06:02 浏览: 52
这段代码是在Python中使用OS模块和open函数,创建一个文件路径为C:\Users\huawei\Desktop\pythonProject1\data\train的目录对象。然后使用sort函数对r'C:\Users\huawei\Desktop\pythonProject1\data\train'目录下的文件名进行排序,按照文件名前面的数字从小到大排序。最后将排序后的文件列表写入一个名为train_list.txt的文本文件中。