批量读取文件夹中的图片，并把图片名写入txt文档中

批量读取文件夹中的图片，并把图片名写入txt文档中

import os

file_path = "./workers"
data_dir = r"E:\study\Head_Pose_Evaluation\datasets"
txt_save_path = "E:/study/Head_Pose_Evaluation/datasets/file.txt"

# 打开图片列表清单txt文件
fw = open(txt_save_path, "w")
# workers文件夹的绝对路径
path = os.path.join(data_dir, file_path[2:])
path = path.replace("\\", "/")
# 查看文件夹下的图片名称
images_name = os.listdir(path)
# 遍历所有文件名
for eachName in images_name:
    # print(eachName)
    file_name = eachName.split('.')[0]  # 只取图片名称，不取扩展名
    # print(file_name)
    fw.write('./'+file_name+'\n')  # 把图片名称写入file.txt文件中
print("生成打印成功")
fw.close()

# print(images_name)