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

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

码农世界 2024-05-28 前端 99 次浏览 0个评论

用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)

转载请注明来自码农世界,本文标题:《删除实例分割中的特定标签》

百度分享代码,如果开启HTTPS请参考李洋个人博客
每一天,每一秒,你所做的决定都会改变你的人生!

发表评论

快捷回复:

评论列表 (暂无评论,99人围观)参与讨论

还没有评论,来说两句吧...

Top