读取文件夹中的多个文件并将边界框预测保存在CSV文件中

import os
import csv

# 文件夹路径
folder_path = "path/to/folder"

# 保存CSV文件的路径
csv_file_path = "path/to/save.csv"

# 创建CSV文件并写入表头
with open(csv_file_path, 'w', newline='') as csv_file:
    writer = csv.writer(csv_file)
    writer.writerow(["文件名", "边界框"])

    # 遍历文件夹中的每个文件
    for filename in os.listdir(folder_path):
        file_path = os.path.join(folder_path, filename)

        # 读取文件内容
        # 进行边界框预测
        # 将预测的边界框信息保存在CSV文件中
        # 例如：writer.writerow([filename, bounding_box])

# 完成保存