快速生成路径索引文件（.txt格式）

此代码记录使用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!")