[update] adding log capacity on IoU_yolo.py

IoU_yolo.py

@@ -1,8 +1,13 @@
 import os
+import sys 
 import numpy as np
 
-GT_DIR = "test_images/wit396_pdf545/labels"
-PRED_DIR = ""
+
+# GT_DIR = "test_images/wit396_pdf545/labels"
+GT_DIR = sys.argv[1]
+
+# PRED_DIR = "test/yolo8Lg-17-07-2026/labels"
+PRED_DIR = sys.argv[2]
  
 def yolo_to_corners(box):
     """Convertit (x_center, y_center, w, h) en (x1, y1, x2, y2)."""
@@ -89,7 +94,7 @@ def match_boxes(gt_boxes, pred_boxes):
 gt_files = sorted(f for f in os.listdir(GT_DIR) if f.endswith(".txt"))
  
 all_ious = []
- 
+
 for filename in gt_files:
     gt_path = os.path.join(GT_DIR, filename)
     pred_path = os.path.join(PRED_DIR, filename)
@@ -102,9 +107,22 @@ for filename in gt_files:
     all_ious.extend(ious)
  
     mean_iou = np.mean(ious) if ious else 0.0
-    print(f"{filename}: {len(matches)} paires, IoU moyen = {mean_iou:.4f}, "
-          f"GT manqués = {len(unmatched_gt)}, faux positifs = {len(unmatched_pred)}")
- 
-print(f"Fichiers traités : {len(gt_files)}")
-print(f"Paires appariées : {len(all_ious)}")
-print(f"IoU moyen global : {np.mean(all_ious) if all_ious else 0.0:.4f}")
+    message_paires = f"{filename}: {len(matches)} paires, IoU moyen = {mean_iou:.4f}, "
+    message_erreurs = f"GT manqués = {len(unmatched_gt)}, faux positifs = {len(unmatched_pred)}"
+    print(f"{message_paires}\n{message_erreurs}\n")
+    
+    base_dir = PRED_DIR.split('/')
+    base_dir = f"{base_dir[0]}/{base_dir[1]}"
+    with open(f"{base_dir}/log.txt","w") as f:
+        f.write(f"{message_paires}\n{message_erreurs}")
+    
+ 
+fichiers_traités = f"Fichiers traités : {len(gt_files)}"
+print(fichiers_traités)
+paires_appariées = f"Paires appariées : {len(all_ious)}"
+print(paires_appariées)
+IoUmoyenglobal = f"IoU moyen global : {np.mean(all_ious) if all_ious else 0.0:.4f}"
+print(IoUmoyenglobal)
+
+with open(f"{base_dir}/log.txt","a") as f:
+        f.write(f"\n{fichiers_traités}\n{paires_appariées}\n{IoUmoyenglobal}")