updaaaaaaaaaaate

configs/local/strategy_generate_config.yaml

@@ -11,7 +11,7 @@ runner:
     root_dir: "experiments"
       
   generate:
-    voxel_threshold: 0.01
+    voxel_threshold: 0.003
     soft_overlap_threshold: 0.3
     hard_overlap_threshold: 0.6
     filter_degree: 75
@@ -21,15 +21,15 @@ runner:
     save_best_combined_points: False
     save_mesh: True
     overwrite: False
-    seq_num: 10
+    seq_num: 15
     dataset_list:
       - OmniObject3d
   
 datasets:
     OmniObject3d:
       #"/media/hofee/data/data/temp_output"
-      root_dir: "/media/hofee/data/data/box_output" 
-      model_dir: "/media/hofee/data/data/box_object_meshes"
+      root_dir: /media/hofee/repository/full_data_output 
+      model_dir: /media/hofee/data/data/scaled_object_meshes
       from: 0
       to: -1 # -1 means end
       #output_dir: "/media/hofee/data/data/label_output"

preprocess/preprocessor.py

@@ -67,7 +67,7 @@ def save_scene_data(root, scene, scene_idx=0, scene_total=1,file_type="txt"):
     target_mask_label = (0, 255, 0, 255)
     display_table_mask_label=(0, 0, 255, 255)
     random_downsample_N = 32768
-    voxel_size=0.002
+    voxel_size=0.003
     filter_degree = 75
     min_z = 0.2
     max_z = 0.5
@@ -135,7 +135,7 @@ def save_scene_data(root, scene, scene_idx=0, scene_total=1,file_type="txt"):
 
 if __name__ == "__main__":
     #root = "/media/hofee/repository/new_data_with_normal"
-    root = r"/media/hofee/data/data/box_output"
+    root = r"/media/hofee/repository/full_data_output"
     # list_path = r"/media/hofee/repository/full_list.txt"
     # scene_list = []
     
@@ -144,7 +144,7 @@ if __name__ == "__main__":
     #         scene_list.append(line.strip())
     scene_list = os.listdir(root)
     from_idx = 0 # 1000
-    to_idx = len(scene_list) # 1500
+    to_idx = 700 # 1500
     print(scene_list)
 
 
@@ -153,7 +153,7 @@ if __name__ == "__main__":
     total = to_idx - from_idx
     for scene in scene_list[from_idx:to_idx]:
         start = time.time()
-        save_scene_data(root, scene, cnt, total, file_type="txt")
+        save_scene_data(root, scene, cnt, total, file_type="npy")
         cnt+=1
         end = time.time()
         print(f"Time cost: {end-start}")

runners/strategy_generator.py

@@ -102,7 +102,7 @@ class StrategyGenerator(Runner):
         seq_num = min(self.seq_num, non_zero_cnt)
         init_view_list = []
         idx = 0
-        while len(init_view_list) < seq_num:
+        while len(init_view_list) < seq_num and idx < len(pts_list):
             if pts_list[idx].shape[0] > 100:
                 init_view_list.append(idx)
             idx += 1