For循环遍历图像文件夹并输出到单个JSON文件中

import os
import json

def traverse_image_folder(folder_path):
    image_files = []
    for root, dirs, files in os.walk(folder_path):
        for file in files:
            if file.endswith('.jpg') or file.endswith('.png'):
                file_path = os.path.join(root, file)
                image_files.append({
                    'file_name': file,
                    'file_path': file_path
                })
    return image_files

folder_path = 'path/to/image/folder'
image_files = traverse_image_folder(folder_path)

json_data = json.dumps(image_files, indent=4)

with open('output.json', 'w') as file:
    file.write(json_data)