遍历文件夹所有文件，并获得其路径写入txt文件中

import os

# 设置要遍历的文件夹路径和要保存路径的txt文件路径
folder_path = r'D:\XM\yolov7-main\voc\valid\images'  # 更改为要遍历的文件夹的路径
output_file = 'valid.txt'  # 更改为要保存路径的txt文件的路径和名称

# 支持的图片格式
image_extensions = ['.jpg', '.jpeg', '.png', '.bmp', '.gif']

# 使用with语句打开txt文件，并指定'w'模式表示要写入文件
with open(output_file, 'w') as f:
    # 使用os.walk函数遍历文件夹下的所有文件和文件夹
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            # 检查文件是否是图片文件
            if any(ext in file.lower() for ext in image_extensions):
                # 构建图片文件的完整路径，并将其写入到txt文件中
                file_path = os.path.join(root, file)
                f.write(file_path + '\n')

# 关闭txt文件
f.close()