一、代码

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))

二、提取效果