获取文件夹下图片的名字保存为txt文件

import os
import randomxmlfilepath = 'E:\\LT0517\\label2'   #注意这里是image而非mask
txtsavepath = 'E:\\LT0517'
total_xml = os.listdir(xmlfilepath)num=len(total_xml)
list = range(num)trainval = random.sample(list, num)  os.chdir('E:\\LT0517')   ftrainval = open('name.txt', 'w')  # chuang jian yi ge txt wen jianfor i in list :name =total_xml[i][:-4] + '\n'ftrainval.write(name)
ftrainval.close()