此代码记录使用python快速生成文件夹中所有文件的路径索引文件
import os
# 图片文件夹索引
root_dir = r"E:\ImgPrograssing\python\yolov7\Apple_yolo\test"
# 输出地址
dst = r"E:\ImgPrograssing\python\yolov7\Apple_yolo\test_img_list.txt"
f = open(dst, "w+")
id_list = os.listdir(root_dir)
num = len(id_list)
count = 0 # 记录文件数
for id in id_list:
print("id:\n", id)
img_dir = root_dir + id # 文件名
print("img_dir:\n", img_dir)
f.write(img_dir + "\n")
print("图片路径:", img_dir + "\n")
count = count+1
print("已完成输出个数:" + str(count) + "\n")
f.close()
print("Finished!")