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

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

糯米丸子TAT

6441人浏览 · 2021-10-26 20:47:42

糯米丸子TAT · 2021-10-26 20:47:42 发布

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