21 lines
625 B
Python
21 lines
625 B
Python
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}") |