labelme 分割json 文件删除节点操作,保存更新中文和换行问题

# !/usr/bin/env python
# -*- encoding: utf-8 -*-

import os
import json

# 这里写你自己的存放照片和json文件的路径
json_dir = 'F:\\data\\ABD\\liver\\2\\paddle_seg_format\\jsons'
json_files = os.listdir(json_dir)


def d_label():
    # 这里写你要删除的标签名
    delete_name = "nozzle"

    for json_file in json_files:
        
        json_file_ext = os.path.splitext(json_file)
        
        if json_file_ext[1] == '.json':
            # 判断是否为json文件
            jsonfile = json_dir + '\\' + json_file
            
            with open(jsonfile, 'r', encoding='utf-8') as jf:
                info = json.load(jf)
                
                for i, label in enumerate(info['shapes']):
                    if info['shapes'][i]['label'] == delete_name:
                        del info['shapes'][i]
                        # 找到位置进行删除
                # 使用新字典替换修改后的字典
                json_dict = info
            
            # 将替换后的内容写入原文件
            with open(jsonfile, 'w') as new_jf:
                json.dump(json_dict, new_jf)
    
    print('delete label over!')


def d_label2():
    # 删除label 对应的 point <=2的节点  检测框的label 删除
    # 这里写你要删除的标签名
    delete_name = "nozzle"
    a=set()
    b= {'cat':0, 'play':0, 'sky':0, 'water':0}
    for json_file in json_files:
        json_file_ext = os.path.splitext(json_file)
    
        if json_file_ext[1] == '.json':
            # 判断是否为json文件
            jsonfile = json_dir + '\\' + json_file
   

            with open(jsonfile, 'r', encoding='utf-8') as jf:
                info = json.load(jf)
            
                for i, label in enumerate(info['shapes']):
                    points = info['shapes'][i]['points']
                    points_list = [tuple(point) for point in points]
              
                    a.add(info['shapes'][i]['label'])
                    b[info['shapes'][i]['label']]=b[info['shapes'][i]['label']]+1
                    if len(points_list) <=3:
                        print(info['shapes'][i]['label'], len(points_list))
                        del info['shapes'][i]  #直接删除这个字典标签
                        # 找到位置进行删除
                # 使用新字典替换修改后的字典
                json_dict = info
            # !/usr/bin/env python
            # -*- coding: utf-8 -*-

                # 将替换后的内容写入原文件
            with open(jsonfile, 'w',encoding="utf-8") as new_jf:
                json.dump(json_dict, new_jf,ensure_ascii = False,indent=1)

    print('delete label over!')
    print(a)
    for k,v in b.items():
        print(k,v)
  json.dump(json_dict, new_jf,ensure_ascii = False,indent=1)
避免中文乱码和没有换行
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值