csv文件转换yolo格式

最新推荐文章于 2024-05-06 07:25:20 发布

梣杉

最新推荐文章于 2024-05-06 07:25:20 发布

阅读量688

点赞数 6

文章标签： YOLO python 人工智能

本文链接：https://blog.csdn.net/weixin_46076409/article/details/137075879

版权

CSV转yolo

Python代码

import os
import csv

def convert_csv_to_yolo(csv_file, output_dir):
    with open(csv_file, 'r') as file:
        reader = csv.DictReader(file)
        for row in reader:
            filename = row['filename']
            width = int(row['width'])
            height = int(row['height'])
            class_label = row['class']
            xmin = int(row['xmin'])
            ymin = int(row['ymin'])
            xmax = int(row['xmax'])
            ymax = int(row['ymax'])

            # 计算YOLO坐标
            x_center = (xmin + xmax) / (2 * width)
            y_center = (ymin + ymax) / (2 * height)
            bbox_width = (xmax - xmin) / width
            bbox_height = (ymax - ymin) / height

            # 构建YOLO格式字符串
            yolo_line = f"{class_label} {x_center} {y_center} {bbox_width} {bbox_height}\n"

            # 构建输出文件路径
            output_file = os.path.join(output_dir, filename + '.txt')

            # 将YOLO格式字符串写入文件
            with open(output_file, 'a') as output:
                output.write(yolo_line)

def batch_convert_csv_to_yolo(csv_folder, output_dir):
    csv_files = [f for f in os.listdir(csv_folder) if f.endswith('.csv')]
    for csv_file in csv_files:
        csv_file_path = os.path.join(csv_folder, csv_file)
        print(f"Converting {csv_file} to YOLO format:")
        convert_csv_to_yolo(csv_file_path, output_dir)
        print()

# 指定包含CSV文件的文件夹路径
csv_folder_path = ''

# 指定输出文件夹路径
output_folder_path = ''

# 执行批量转换
batch_convert_csv_to_yolo(csv_folder_path, output_folder_path)