python将一个文件中的所有图片的名称放入一个.txt文件中

#开发时间：2023/3/22 10:00
#开发人员：h**
import os
dir1='F:/2022study/yolo/dataset/train/'#图片文件存放地址
txt1 = 'F:/2022study/yolo/dataset/train.txt'#图片文件名存放txt文件地址
f1 = open(txt1,'a')#打开文件流
for filename in os.listdir(dir1):
    f1.write(filename)
    f1.write("\n")#换行
f1.close()#关闭文件流