一、代碼

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

二、提取效果