labeml转yolo数据集格式之实例分割

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


# 参考链接 https://blog.csdn.net/m0_63330473/article/details/135079898
# 适应场景:lableme标注软件生成的json实例分割标签转yolo适用的txt格式


def json_to_txt(jsonfilePath, resultDirPath, ):
    """
    jsonfilePath: labelme标注好的*.json文件所在文件夹
    resultDirPath: 转换好后的*.txt保存文件夹
    classList: 数据集中的类别标签
    """

    class_names = []

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

    # for jsonfile in jsonfileList:
    # for jsonfile in jsonfileList:
    for jsonfile in tqdm(jsonfileList, desc='Processing'):

        with open(jsonfile, "r", encoding='UTF-8') as f:
            file_in = json.load(f)

            # 4. 读取文件中记录的所有标注目标
            shapes = file_in["shapes"]

            # 5. 使用图像名称创建一个txt文件,用来保存数据
            with open(resultDirPath + "\\" + jsonfile.split("\\")[-1].replace(".json", ".txt"), "w") as file_handle:
                # 6. 遍历shapes中的每个目标的轮廓
                for shape in shapes:
                    if shape["label"] not in class_names:
                        class_names.append(shape["label"])
                    # 7.根据json中目标的类别标签,从classList中寻找类别的ID,然后写入txt文件中
                    file_handle.writelines(str(class_names.index(shape["label"])) + " ")

                    # 8. 遍历shape轮廓中的每个点,每个点要进行图像尺寸的缩放,即x/width, y/height
                    for point in shape["points"]:
                        x = point[0] / file_in["imageWidth"]  # mask轮廓中一点的X坐标
                        y = point[1] / file_in["imageHeight"]  # mask轮廓中一点的Y坐标
                        file_handle.writelines(str(x) + " " + str(y) + " ")  # 写入mask轮廓点

                    file_handle.writelines("\n")
            file_handle.close()
        f.close()
    with open(resultDirPath + '\\' + 'classes.txt', 'w') as f:
        for i in class_names:
            f.write(i + '\n')


if __name__ == "__main__":
    jsonfilePath = r"C:\Users\18703\Desktop\data\glue2\jiaojson"  # 要转换的json文件所在目录
    resultDirPath = r"C:\Users\18703\Desktop\data\glue2\txt"  # 要生成的txt文件夹
    json_to_txt(jsonfilePath=jsonfilePath, resultDirPath=resultDirPath, )

  • 5
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值