精灵标注助手json转csv

橘座的忧伤

于 2020-05-07 16:42:10 发布

阅读量450

点赞数

CC 4.0 BY-SA版权

分类专栏：数据处理

本文链接：https://blog.csdn.net/qq_43100178/article/details/105975869

数据处理专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一个使用Python脚本将JSON格式的目标检测数据转换为YOLOv3所需的训练格式的过程。该脚本读取指定目录下的JSON文件，解析路径、尺寸和边界框信息，并按比例缩放至YOLO所需的格式。

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

import os
import cv2
import numpy as np
import json
from cfg import *

with open(LABEL_FILE, "w+") as f:
    for json_file in os.listdir("F:\A_I\myyolo3\data\label"):
        with open(os.path.join("F:\A_I\myyolo3\data\label", json_file), "r") as jf:
            j = json.load(jf)
            print(j)
            pic = j['path'].split("\\")[-1]
            boxes = j['outputs']['object']
            wh = j['size']
            w, h = wh['width'], wh['height']

            w_scale, h_scale = w / IMG_WIDTH, h / IMG_HEIGHT

            f.write(pic)
            for box in boxes:
                bndbox = box['bndbox']
                _x1, _y1, _x2, _y2 = bndbox['xmin'], bndbox['ymin'], bndbox['xmax'], bndbox['ymax']
                _w, _h = _x2 - _x1, _y2 - _y1
                _w0_5, _h0_5 = _w / 2, _h / 2
                _cx, _cy = _x1 + _w0_5, _y1 + _h0_5
                x1, y1, w, h = int(_cx / w_scale), int(_cy / h_scale), int(_w / w_scale), int(_h / h_scale)
                f.write(f" {int(box['name'])} {x1} {y1} {w} {h}")
            f.write("\n")
            f.flush()