python批量修改labelme json格式标签

之前打算用mask-rcnn跑一个数据集,标注方式出点问题   使用labelname1,labelname2这种的标注模式  现在要改成labelname   group_id:1这种,还不是自己标的也不好意思在让别人去改     图片太多  手工一个。先能用  后期再优化一下。用上正则和多线程  还有CPython

以下是代码  

import json
import os

path = './'
dirs = os.listdir(path)



def dateset():
    num_flag = 0
    for file in dirs: # 循环读取路径下的文件并筛选输出
        if os.path.splitext(file)[1] == ".json": # 筛选图片文件
            num_flag = num_flag +1
    
            print("path ===== ",file)
            # print(os.path.join(path,file))
            with open(os.path.join(path,file),'r') as load_f:
    
                load_dict = json.load(load_f)
            n=len(load_dict['shapes'])
           
            for i in range (0,n):
                
                ###钢筋###
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks1':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '1'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks2':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '2'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks3':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '3'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks4':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '4'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks5':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '5'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks6':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '6'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks7':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '7'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks8':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '8'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks9':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '9'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks10':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '10'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks11':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '11'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks12':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '12'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks13':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '13'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks14':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '14'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks15':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '15'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks16':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '16'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks17':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '17'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks18':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '18'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks19':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '19'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks20':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '1'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks1':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '20'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks21':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '21'
                
                if load_dict['shapes'][i]['label'] == 'flat_bar_with_hooks22':
                    load_dict['shapes'][i]['label'] = 'flat_bar_with_hooks'
                    load_dict['shapes'][i]['group_id'] = '22'
                
                
                  ###板###
                
                
                if load_dict['shapes'][i]['label'] == 'concrete_plate1':
                    load_dict['shapes'][i]['label'] = 'concrete_plate'
                    load_dict['shapes'][i]['group_id'] = '1'
                
                if load_dict['shapes'][i]['label'] == 'concrete_plate2':
                    load_dict['shapes'][i]['label'] = 'concrete_plate'
                    load_dict['shapes'][i]['group_id'] = '2'
                
                if load_dict['shapes'][i]['label'] == 'concrete_plate3':
                    load_dict['shapes'][i]['label'] = 'concrete_plate'
                    load_dict['shapes'][i]['group_id'] = '3'
                
                if load_dict['shapes'][i]['label'] == 'concrete_plate4':
                    load_dict['shapes'][i]['label'] = 'concrete_plate'
                    load_dict['shapes'][i]['group_id'] = '4'
                
                if load_dict['shapes'][i]['label'] == 'concrete_plate5':
                    load_dict['shapes'][i]['label'] = 'concrete_plate'
                    load_dict['shapes'][i]['group_id'] = '5'
                
                if load_dict['shapes'][i]['label'] == 'concrete_plate6':
                    load_dict['shapes'][i]['label'] = 'concrete_plate'
                    load_dict['shapes'][i]['group_id'] = '6'
                    
                    
                    
                    ###线盒###
                    
                
                if load_dict['shapes'][i]['label'] == 'plastic_box1':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '1'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box2':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '2'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box3':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '3'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box4':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '4'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box5':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '5'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box6':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '6'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box7':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '7'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box8':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '8'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box9':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '9'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box10':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '10'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box11':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '11'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box12':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '12'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box13':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '13'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box14':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '14'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box15':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '15'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box16':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '16'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box17':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '17'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box18':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '18'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box19':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '19'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box20':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '20'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box21':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '21'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box22':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '22'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box23':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '23'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box24':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '24'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box25':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '25'
                
                if load_dict['shapes'][i]['label'] == 'plastic_box26':
                    load_dict['shapes'][i]['label'] = 'plastic_box'
                    load_dict['shapes'][i]['group_id'] = '26'
                    
                    ###桁架筋
                    ###
                
                if load_dict['shapes'][i]['label'] == 'truss_bars1':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '1'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars2':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '2'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars3':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '3'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars4':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '4'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars5':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '5'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars6':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '6'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars7':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '7'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars8':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '8'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars9':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '9'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars10':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '10'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars11':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '11'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars12':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '12'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars13':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '13'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars14':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '14'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars15':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '15'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars16':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '16'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars17':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '17'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars18':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '18'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars19':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '19'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars20':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '20'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars21':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '21'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars22':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '22'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars23':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '23'
                
                if load_dict['shapes'][i]['label'] == 'truss_bars24':
                    load_dict['shapes'][i]['label'] = 'truss_bars'
                    load_dict['shapes'][i]['group_id'] = '24'
                
                
                with open(os.path.join(path,file),'w') as dump_f:
                    json.dump(load_dict, dump_f)









dateset()







简单粗暴  if的让我都不好意思。  不过比较容易看明白替换成自己的需求吧(弱弱安慰自己一下)。

你可能感兴趣的:(实例分割)