将一个文件夹下的所有图片的名字存到txt文件_AWhiteDongDong的博客

首先在文件夹下先创建一个txt文件，然后执行以下代码：

import os

img_path = './测试集/'
img_list = os.listdir(img_path)
f = open('test.txt', 'w')
for img in img_list:
    img = img[:-4] + '\n'
    f.write(img)
f.close()

import os