将图片名批量读取并存到.txt文件中

博客参考：https://blog.csdn.net/qq_34585338/article/details/90299401

import os
import random
 
image_path = r'C:\Users\jiabin.ouyang\Desktop\BXS'   # 这里是image文件夹路径
total_xml = os.listdir(image_path)
 
num=len(total_xml)
list = range(num)
trainval = random.sample(list, num)  
 
os.chdir(r'C:\Users\jiabin.ouyang\Desktop\to')   
ftrainval = open('name.txt', 'w')  # 将存入数据的文本取名为：name.txt
 
for i in list :
  # name =total_xml[i][:-4] + '\n'  
  # 去除后面的四个字符，因此去掉了后面的.bmp，即为舍去图片的后缀名.
  name =total_xml[i] + '\n'
  ftrainval.write(name)
ftrainval.close()