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

# --把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]) / 
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
将实例分割的json文件转换为yolotxt文件的主要步骤如下: 1. 读取json文件中的数据,包括图片大小、目标数量、目标类别、目标位置等信息。 2. 将目标位置信息按照yolo的格式进行转换。yolo格式中的坐标是归一化的,即左上角坐标为(0,0),右下角坐标为(1,1)。 3. 将目标类别信息转换为对应的数字标签。 4. 将转换后的目标信息按照yolo的格式写入到txt文件中。 下面是一个基于Python语言的代码示例,用于将json文件转换为yolotxt文件: ```python import json # 定义目标类别和对应数字标签的映射关系 class_map = {'car': 0, 'person': 1, 'dog': 2} def convert_json_to_txt(json_file, txt_file): with open(json_file, 'r') as f: data = json.load(f) img_width = data['imageWidth'] img_height = data['imageHeight'] objects = data['objects'] num_obj = len(objects) with open(txt_file, 'w') as f: for obj in objects: class_name = obj['classTitle'] if class_name not in class_map: continue class_id = class_map[class_name] x, y, w, h = obj['points']['exterior'][0] x_min = x / img_width y_min = y / img_height x_max = (x + w) / img_width y_max = (y + h) / img_height f.write(f"{class_id} {x_min:.6f} {y_min:.6f} {x_max:.6f} {y_max:.6f}\n") ``` 在上述代码中,`json_file`是要转换的json文件路径,`txt_file`是输出的txt文件路径。`class_map`定义了目标类别和对应数字标签的映射关系,可以根据实际情况进行修改。`convert_json_to_txt`函数读取json文件中的数据,将目标位置信息按照yolo的格式进行转换,并将转换后的目标信息写入到txt文件中。最后,调用该函数即可完成json文件yolotxt文件的转换。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值