添加状态分类和液面分割

yolo11_point/trains.py

@@ -0,0 +1,21 @@
+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}")