之前打算用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的让我都不好意思。 不过比较容易看明白替换成自己的需求吧(弱弱安慰自己一下)。