diff --git a/main.py b/main.py index 9684bb75465b755337541a099157cae7031e8ef3..cced98fe248e6fa9029e7ce80058911fe00e9bb2 100644 --- a/main.py +++ b/main.py @@ -29,8 +29,7 @@ def generate_folders(name, list_categories): if not is_exist2: os.makedirs(f"{name}/Generated/{f}/{cat}") -if __name__ == '__main__': - +if __name__ == '__main__': # Create the parser parser = argparse.ArgumentParser() # Add an argument @@ -39,23 +38,22 @@ if __name__ == '__main__': # Parse the argument args = parser.parse_args() - ### parameters ### Categories = [] # to read Nb_instance = 1 occ_target = 0.5 - dataset_src = "/media/gduret/DATA/dataset/s2rg/Fruits_all_medium/data" + dataset_src = "/gpfsscratch/rech/uli/ubn15wo/dataset/s2rg/Fruits_all_medium/data/" #dataset_src = "/media/mahmoud/E/Fruits_easy/data" choice = "low" # depth of rgb resolution datas data_options = {"high": "ground_truth_rgb", "low": "ground_truth_depth"} dataset_type = data_options[choice] - dataset_name = f"GUIMOD_{choice}" + dataset_name = f"/gpfsscratch/rech/uli/ubn15wo/dataset/s2rg/Fruits_all_medium/GUIMOD_{choice}" list_categories = ["banana1", "kiwi1", "pear2", "strawberry1", "apricot", "orange2", "peach1", "lemon2", "apple2" ] # frame = "1_600000000" #frame = "1_926000000" Nb_camera = 15 - #Nb_world = 2 + #Nb_world = 10000 generate_folders(dataset_name, list_categories) for cat in list_categories: @@ -69,7 +67,7 @@ if __name__ == '__main__': #print("pcd", pcd) - fps_points = apply_fps(pcd, 200) + fps_points = apply_fps(pcd, 8) #print(fps_points) np.savetxt(f'{dataset_name}/Generated/FPS/{categories}_fps_3d.txt', fps_points)