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()
フォルダー内のすべてのファイルを走査し、パスを取得して txt ファイルに書き込みます。
おすすめ
転載: blog.csdn.net/qq_54372122/article/details/130195969
おすすめ
ランキング