将labelme勾画的json格式标签转化YOLO的txt格式标签

Admiral~

已于 2022-10-08 14:31:48 修改

阅读量852

点赞数

文章标签： json python 深度学习

于 2022-10-08 13:05:35 首次发布

本文链接：https://blog.csdn.net/Admiral_x/article/details/127206729

版权

1、完整代码

import os
import json
import time
import numpy as np
from tqdm import tqdm
from threading import Thread


def ConvertOne(labelme_dir, save_dir, json_file, classes, del_class):
    json_file_path = os.path.join(labelme_dir, json_file)
    yolo_save_path = os.path.join(save_dir, json_file.replace(".json", ".txt"))
    yolo_label_file = open(yolo_save_path, "w+")

    with open(json_file_path, "r") as jf:
        labelme_labels = json.load(jf)
        # # 根据标注时的形状取出标签
        # polygons =