根据文件内的图片生成对应的txt索引目录

一个文件有大量的图片，可以通过下面的代码，将该文件夹下的所有文件名生成一个txt文档，并使用换行隔开。

import os

# 设置文件夹路径
folder_path = '你的文件夹路径'  # 替换为你的文件夹路径

# 获取文件夹名称
folder_name = os.path.basename(folder_path)

# 设置输出txt文件路径
txt_file_path = os.path.join(folder_path, f'{folder_name}.txt')

# 获取文件夹中的所有图片文件名
image_extensions = ('.png', '.jpg', '.jpeg', '.gif', '.bmp')  # 可根据需要添加其他格式
image_names = [file for file in os.listdir(folder_path) if file.lower().endswith(image_extensions)]

# 将文件名写入txt文件
with open(txt_file_path, 'w') as txt_file:
    for image_name in image_names:
        txt_file.write(f"{image_name}\n")

print(f"图片文件名已写入 {txt_file_path}。")