yolov5训练json转txt

在 YOLOv5 中，标注文件通常使用的是 .txt 格式，每个 .txt 文件对应一张图片，其中每一行表示一个物体的标注信息。对于 COCO 格式的标注文件，需要将其转换为 YOLOv5 所需要的 .txt 格式。下面是一个将 COCO 格式的标注文件转换为 YOLOv5 格式的 Python 代码示例。

import json

# COCO 类别编号与 YOLOv5 类别编号的对应关系
class_mapping = {
    0: 0,  # "person": "person"
    1: 1,  # "bicycle": "bicycle"
    2: 2,  # "car": "car"
    # ...
}

def convert_coco_to_yolov5(coco_path, yolov5_path):
    with open(coco_path, 'r') as f:
        coco_data = json.load(f)
    
    with open(yolov5_path, 'w') as f:
        for image in coco_data['images']:
            image_id = image['id']
            image_width = image['width']
            image_height = image['height']
            for annotation in coco_data['annotations']:
                if annotation['image_id'] == image_id:
                    class_id = class_mapping[annotation['category_id']]
                    bbox = annotation['bbox']
                    x_center = bbox[0] + bbox[2] / 2
                    y_center = bbox[1] + bbox[3] / 2
                    width = bbox[2]
                    height = bbox[3]
                    x_center /= image_width
                    y_center /= image_height
                    width /= image_width
                    height /= image_height
                    line = f"{class_id} {x_center} {y_center} {width} {height}\n"
                    f.write(line)

在上面的代码中，`coco_path` 表示 COCO 格式的标注文件路径，`yolov5_path` 表示转换后的 YOLOv5 格式的标注文件路径。`class_mapping` 是一个字典，用于将 COCO 类别编号映射为 YOLOv5 类别编号。在 `convert_coco_to_yolov5` 函数中，首先读取 COCO 格式的标注文件，然后遍历所有图片和标注信息，将每个物体的标注信息转换为 YOLOv5 格式，并写入到对应的 .txt 文件中。