标注错误后,批量修改labelme格式标签

import os
import json    
def get_car_json_info(data_dir):
    all_json=os.listdir(data_dir)
    print(len(all_json))
    trucklist = [ "car", "truck"]  # 正确标签,注意顺序
    for j_name in all_json:
        with open(data_dir + j_name,'r',encoding='utf-8') as j:
            info=json.load(j)
            # print(j_name)
            slist = info["shapes"]
            # 查找错误标签
            for i in range(len(slist)):
                # print(slist[i]["label"])
                if slist[i]["label"] in trucklist :
                    continue
                else:
                    print(j_name)
            # 修改car 为 Car
            for i in range(len(slist)):
                # print(slist[i]["label"])
                if slist[i]["label"] == "bus":
                    info["shapes"][i]["label"] = "car"
                with open(data_dir + j_name,'w',encoding='utf-8') as f:
                    json.dump(info, f, ensure_ascii=False)

get_car_json_info("F:/ew/dataset_new/Annotations/")

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值