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