Python批量提取文件夹中图片的名称及路径到指定的.txt文件中

一、代码

import os

# 定义要保存的文件名
file_name = "TestImage/Image_Visible_Gray.txt"

# 读取文件夹路径
folder_path = "TestImage/Image_Visible_Gray"

# 遍历文件夹中的所有文件
with open(file_name, "w") as f:
  for filename in os.listdir(folder_path):
    # 判断是否是图片文件
    if os.path.isfile(os.path.join(folder_path, filename)) and filename.endswith((".jpg", ".png", ".jpeg")):
      # 将图片名写入文件
      f.write("/home/zuoka/Code/Python/RknnToolkit_1/RknnToolkit/ecbsr_1.7.5/TestImage_Visible/Image_Visible_Gray/"+filename + "\n")

print("已将所有图片名保存到文件 {} 中".format(file_name))

二、提取效果