批量读取文件夹中的图片,并把图片名写入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)