删除实例分割中的特定标签

用labelme软件对图像进行实例分割或语义分割标注后会得到json文件，如果想要删除某个特定标签，可以使用如下代码，完整代码下载地址：代码地址

import json
import os

# 要处理的json文件夹路径
folder_path = 'H:/json'
# 需要删除的标签列表,填写自己的标签
labels_to_remove = ['0', '1', '2', '3']

# 遍历文件夹中的所有文件
for filename in os.listdir(folder_path):
    if filename.endswith('.json'):
        file_path = os.path.join(folder_path, filename)

        # 读取JSON文件
        with open(file_path, 'r') as file:
            data = json.load(file)

        # 遍历并删除特定标签的标注
        data['shapes'] = [annotation for annotation in data['shapes'] if annotation['label'] not in labels_to_remove]

        # 保存修改后的JSON文件
        with open(file_path, 'w') as file:
            json.dump(data, file, indent=4)