将labelme标注的json文件转换成txt文件

最新推荐文章于 2024-08-05 08:30:00 发布

不会写代码的ppl

最新推荐文章于 2024-08-05 08:30:00 发布

阅读量1.9k

点赞数

分类专栏：计算机视觉文章标签： python json 深度学习机器学习神经网络

本文链接：https://blog.csdn.net/qq_37973756/article/details/115932368

版权

本文介绍如何利用Python将LabelMe工具生成的JSON格式标注数据转换为适用于深度学习和机器学习任务的TXT文件。通过解析JSON文件，提取关键标注信息并保存到TXT中，简化数据预处理流程。

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

import json
import os

"""
Computer: 2
Headdown: 1
Headup:   0
Text:     3
"""


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