json类型文件转为yolo可识别的txt文件

最新推荐文章于 2025-06-12 15:58:31 发布

$ሀIT

最新推荐文章于 2025-06-12 15:58:31 发布

阅读量764

点赞数

CC 4.0 BY-SA版权

文章标签： python 深度学习 pytorch

本文链接：https://blog.youkuaiyun.com/t840703721/article/details/115606980

本文介绍了如何使用Python将JSON文件转换为YOLO（You Only Look Once）算法可识别的TXT标注文件。适用于深度学习和计算机视觉任务，特别是基于PyTorch的项目。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# --把4点标注的json文件转化为（x_c,y_c,h,w,theta)的yolo txt 格式
import json
import os
import math
import random

my_class = ["bottle"]


def convert(image_id):

    in_file = 'annotations/%s.json' % image_id
    out_file = open('labels/%s.txt' % image_id, 'w')
    with open(in_file) as f_obj:
        label = json.load(f_obj)

    for bbox in label["shapes"]:
        vertices = bbox["points"]  # 顶点列表
        a, b, c, d = vertices

        x_c = (a[0]+b[0]+c[0]+d[0])/4
        y_c = (a[1] + b[1] + c[1] + d[1]) / 4