import os
import json

# 设置你的目标文件夹路径
folder_path = "../../data/part5/精标"

# 定义替换映射
replace_dict = {
    "引擎盖": 2,
    "前保险杠": 1,
    "前挡风玻璃": 4,
    "背景": 0,
    "A柱": 3
}

def replace_labels_in_json(file_path):
    with open(file_path, 'r', encoding='utf-8') as f:
        data = json.load(f)

    def replace(obj):
        if isinstance(obj, dict):
            return {k: replace(v) for k, v in obj.items()}
        elif isinstance(obj, list):
            return [replace(item) for item in obj]
        elif isinstance(obj, str) and obj in replace_dict:
            return replace_dict[obj]
        else:
            return obj

    new_data = replace(data)

    with open(file_path, 'w', encoding='utf-8') as f:
        json.dump(new_data, f, ensure_ascii=False, indent=2)

if __name__ == "__main__":
    for filename in os.listdir(folder_path):
        if filename.endswith('.json'):
            file_path = os.path.join(folder_path, filename)
            replace_labels_in_json(file_path)
    print("替换完成！")