tool: counting the number of segments

README.md

@@ -79,3 +79,11 @@ python3 hierarchiqueAvecDerniereDistance.py \
     --output_name_average dendro_avg.png \
     --output_name_complete dendro_complete.png
 ```
+
+# Tools
+
+### • Counting the number of segments
+
+```bash
+python3 count_nb_segments.py /path/to/your/segments
+```

count_nb_segments.py

+import argparse
+import numpy as np
+from pathlib import Path
+
+def main(args):
+    dossier = Path(args.dossier)
+    fichiers_segment = sorted(dossier.glob("**/*_segmentfeature.npy"))
+
+    if not fichiers_segment:
+        print("Erreur.")
+        return
+
+    total_segments = 0
+
+    for fichier in fichiers_segment:
+        try:
+            seg_features = np.load(fichier)
+            n_segments = seg_features.shape[0]
+            total_segments += n_segments
+        except Exception as e:
+            print(f"Erreur.")
+
+    print(f"Nombre total de segments : {total_segments}")
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Comptage des segments dans les fichiers segmentfeature.npy")
+    parser.add_argument("dossier", type=str, help="Chemin vers le dossier contenant les fichiers .npy")
+    args = parser.parse_args()
+
+    main(args)