import os

labels_dir = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/seg/dataset2/labels"

for txt_file in os.listdir(labels_dir):
    if not txt_file.endswith(".txt"):
        continue
    path = os.path.join(labels_dir, txt_file)
    lines = open(path, "r").readlines()
    new_lines = []
    for line in lines:
        parts = line.strip().split()
        if not parts:
            continue
        # 修改类别 ID：1 -> 0
        if parts[0] == "1":
            parts[0] = "0"
        new_lines.append(" ".join(parts))

    open(path, "w").write("\n".join(new_lines))
    print(f"✅ 已处理: {txt_file}")