Idk

rpi/training/decrease_labels.py

@@ -0,0 +1,36 @@
+import os
+
+# --------------------------
+# Configuration
+# --------------------------
+labels_dir = "datasets/visiope/test/labels"
+
+# --------------------------
+# Process each label file
+# --------------------------
+for filename in os.listdir(labels_dir):
+    if not filename.endswith(".txt"):
+        continue
+
+    txt_path = os.path.join(labels_dir, filename)
+    new_lines = []
+
+    with open(txt_path, "r") as f:
+        lines = f.readlines()
+
+    for line in lines:
+        parts = line.strip().split()
+        if len(parts) < 5:
+            continue  # skip invalid lines
+        cls = int(parts[0])
+        cls -= 1  # subtract 1 from class index
+        cls = max(cls, 0)  # ensure no negative indices
+        new_lines.append(" ".join([str(cls)] + parts[1:]))
+
+    # Overwrite file with updated indices
+    with open(txt_path, "w") as f:
+        f.write("\n".join(new_lines))
+
+    print(f"Updated {filename}")
+
+print("All label files have been adjusted!")