python获取文件夹下的所有文件名，并写入txt文件中

需求：如图，在 test 文件夹下编写 python 文件 path_get_to_txt.py ，使 ./images/train 和 ./images/val 下的图片名填入对应的 txt 文件中。

代码如下：

# !/usr/bin/env python
# -*- encoding: utf-8 -*-

import os

# 图片所存放的路径，train 和 val 内的数目相近
train_path = './images/train'                                           # 第1处 需要修改为自己的路径
val_path = './images/val'                                               # 第2处 需要修改为自己的路径

# 获取文件名
train_list = os.listdir(train_path)
val_list = os.listdir(val_path)

# 输出所得到的文件名
# print('img_list: ', img_list)

# 打开 txt 文件，并将路径写入
with open('./train.txt', 'w') as f:                                     # 第3处 需要修改为自己的路径
    for train_name in train_list:
        f.write('./datasets/circle/images/train/'+ train_name + '\n')   # 第4处 需要修改为自己的路径

with open('./val.txt', 'w') as f:                                       # 第5处 需要修改为自己的路径
    for val_name in val_list:
        f.write('./datasets/circle/images/val/'+ val_name + '\n')       # 第6处 需要修改为自己的路径

print('\n' + 'OK')

关于代码的修改处：

• 第1处 和 第2处 为需要获取文件名的文件所在路径
• 第3处 和 第5处 为需要写入的 txt 文件所在路径
• 第4处 和 第6处 为 txt 文件中需要写入的规范形式

成功运行后输出 OK