ISAT类别编辑批处理代码(利用ISAT编辑生成的json文件,对其 合并、删除、保留)

在利用半自动处理工具勾绘的类别,想要对其精选处理,具体示例代码如下(优先级:merge_dict>labels_to_delete>abels_to_delete):

import os
import json
import shutil
from collections import defaultdict

# ====================== 🚩 配置区域 ====================== #
input_dir = r"D:\plant_seg_datasets\20250524\test_img"
output_dir = input_dir + "_edited1"

# 标签合并配置 (旧标签: 新标签)
merge_dict = {
    # 示例:
    # "old_label": "new_label",
}

# 要删除的标签集合
labels_to_delete = set()  # 例如:{"trash", "ignore"}

# 要保留的标签集合 (设为None表示保留所有)
labels_to_keep = {"hei_shui_yin_lian_hua",
                  'duo_bei_yin_lian_hua',
                  'ding_bing_hu',
                  'dan_hua_ji',
                  'dong_bei_yan_hu_suo',
                  'tu_kui',
                  'fan_e_yin_lian_hua'
                  }  # 只保留这个标签


# ======================================================== #

def process_isat_files(input_dir, output_dir):
    """处理ISAT格式的JSON文件"""
    os.makedirs(output_dir, exist_ok=True)

    # 统计计数器
    stats = defaultdict(lambda: defaultdict(int))
    processed_files = 0
    total_json_files = 0
    skipped_files = 0
    found_all_labels = set()  # 收集所有发现的标签

    for filename in os.listdir(input_dir):
        if not filename.endswith(".json"):
            # 复制非JSON文件(如图片)
            src = os.path.join(input
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值