json 转为txt

没事哒

已于 2024-06-27 11:17:23 修改

阅读量848

点赞数 10

文章标签： json 人工智能计算机视觉开发语言

于 2024-01-13 09:29:02 首次发布

本文链接：https://blog.youkuaiyun.com/2301_79485441/article/details/135565747

版权

这段代码是一个Python函数，用于将LabelMe格式的JSON文件转换为YOLOv2所需的.txt格式，用于目标检测。它读取JSON中的形状和点信息，并按照指定的类别列表输出坐标。

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

import json
import os
import glob
import os.path as osp


def labelme2yolov2Seg(jsonfilePath="", resultDirPath="", classList=["buds"]):
    if (not os.path.exists(resultDirPath)):
        os.mkdir(resultDirPath)

    jsonfileList = glob.glob(osp.join(jsonfilePath, "*.json"))

    print(jsonfileList)

    for jsonfile in jsonfileList:
        with open(jsonfile, "r") as f:
            file_in = json.load(f)

            shapes = file_in["shapes"]

            with open(resultDirPath + "\\" + jsonfile.split("\\")[-1].replace(".json", ".txt"), "w") as file_handle:
                for shape in shapes:
                    file_handle.writelines(str(classList.index(shape["label"])) + " ")

                    for point in shape["points"]:
                        x = point[0] / file_in["imageWidth"]