From 0df7d06dcbe1942ff878cc6b5b230b042e42e20e Mon Sep 17 00:00:00 2001
From: gduret <guillaume.duret@ec-lyon.fr>
Date: Wed, 18 Oct 2023 14:13:25 +0000
Subject: [PATCH] compute label for downloaded dataset

---
 compute_label.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/compute_label.py b/compute_label.py
index 3d6d1ce..d6cb4b1 100644
--- a/compute_label.py
+++ b/compute_label.py
@@ -31,7 +31,7 @@ if __name__ == "__main__":
     #path_data = "/gpfsscratch/rech/uli/ubn15wo/dataset1/s2rg/Fruits_all_medium/GUIMOD_low/Generated_Worlds_Training/orange2/Pose_transformed/"
     #path_data = f"/gpfsscratch/rech/uli/ubn15wo/dataset1/s2rg/Fruits_all_medium/GUIMOD_low/{target_folder}/"
 
-    for split in ["_Training", "_Evaluating", "_Testing"]:
+    for split in [""]:
         path_data = f"{args.path_dataset}/{target_folder}{split}"
 
         if not os.path.exists(path_data_df) : 
@@ -75,7 +75,7 @@ if __name__ == "__main__":
             print("-------------------------------------------------------------------------------------------------------")
             print("-------------------------------------------------------------------------------------------------------")
             
-            if split=="_Training" : 
+            if split=="" : 
                 with open(f"{path_data_df}/0{obj_ids[obj]}/train{occ_data}.txt", 'w') as f:
                     for filename2 in os.listdir(path_poses):
                         id_file2=filename2.split('.')[0]
@@ -133,15 +133,15 @@ if __name__ == "__main__":
     for obj in objects:
 
         # Reading data from file1
-        with open(f"{path_data_df}/0{obj_ids[obj]}/gt_{target_folder}_Training{occ_data}.yml") as fp:
+        with open(f"{path_data_df}/0{obj_ids[obj]}/gt_{target_folder}{occ_data}.yml") as fp:
             data_train = fp.read()
         
         # Reading data from file2
-        with open(f"{path_data_df}/0{obj_ids[obj]}/gt_{target_folder}_Evaluating{occ_data}.yml") as fp:
+        with open(f"{path_data_df}/0{obj_ids[obj]}/gt_{target_folder}{occ_data}.yml") as fp:
             data_eval = fp.read()
 
         # Reading data from file2
-        with open(f"{path_data_df}/0{obj_ids[obj]}/gt_{target_folder}_Testing{occ_data}.yml") as fp:
+        with open(f"{path_data_df}/0{obj_ids[obj]}/gt_{target_folder}{occ_data}.yml") as fp:
             data_test = fp.read()
 
         
-- 
GitLab