Python 快速获取文件夹中的所有文件名,并保存到txt文件中

下面的代码可以读取文件夹中的所有文件名,并记录到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')

你可能感兴趣的:(python)