compute label for downloaded dataset

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()