python批量更改实例分割数据集标签

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'] == 'angel':
                    load_dict['shapes'][i]['label'] = 'scale'



                with open(os.path.join(path, file), 'w') as dump_f:
                    json.dump(load_dict, dump_f)


dateset()