将train文件夹下的文件保存到train.txt中,代码如下所示:
import os
path = "D:/Code/Data/roaddata/MassachusettsRoadsDataset/tiff/val"
file_name_list = os.listdir(path)
with open('D:/Code/Data/roaddata/MassachusettsRoadsDataset/list/valid.txt','w') as f:
for file_name in file_name_list:
f.write(file_name[0:-5]+'\n') # -5表示从后往前数,到小数点位置
过程及结果如下: