下面的代码可以读取文件夹中的所有文件名,并记录到txt文件中,可以应用到如深度学习制作数据集等场景中
# !/usr/bin/env python
# -*- encoding: utf-8 -*-
import os
img_path = './images/'
img_list=os.listdir(img_path)
print('img_list: ',img_list)
with open('testImagelist.txt','w') as f:
for img_name in img_list:
f.write(img_name+'\n')