在利用半自动处理工具勾绘的类别,想要对其精选处理,具体示例代码如下(优先级: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

最低0.47元/天 解锁文章
4652

被折叠的 条评论
为什么被折叠?



