Labelme标注文件json批量转txt的python代码

最新推荐文章于 2024-05-20 16:58:48 发布

Ben Dan

最新推荐文章于 2024-05-20 16:58:48 发布

阅读量711

点赞数 2

分类专栏：深度学习文章标签：神经网络机器学习深度学习 json 数据挖掘

本文链接：https://blog.csdn.net/weixin_44574507/article/details/118103449

版权

本文介绍了一个Python代码，用于将使用Labelme工具标注的JSON文件批量转换为YOLO模型所需的TXT格式。代码适用于矩形框标注，转换过程中需修改保存路径、输入JSON文件夹路径以及根据标签类别调整内容。转换后的TXT文件每行包含标签和矩形框坐标信息。

摘要由CSDN通过智能技术生成

将json标注文件转为txt格式

用Labelme标注图片后，需要批量转为YOLO系列模型所需的txt格式
注意，此代码仅适用矩形框标注方法的转换

import json
import os

def convert(img_size, box):
    x1 = box[0]
    y1 = box[1]
    x2 = box[2]
    y2 = box[3]

    center_x = (x1 + x2) * 0.5 / img_size[0]
    center_y = (y1 + y2) * 0.5 / img_size[1]
    w = abs((x2 - x1)) * 1.0 / img_size[0]
    h = abs((y2 - y1)) * 1.0 / img_size[1]

    return (center_x, center_y, w, h)


def decode_json(jsonfloder_path<