ready for jz ?

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)