diff --git a/configs/local/strategy_generate_config.yaml b/configs/local/strategy_generate_config.yaml index 09ccd46..7b88d9c 100644 --- a/configs/local/strategy_generate_config.yaml +++ b/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" diff --git a/preprocess/preprocessor.py b/preprocess/preprocessor.py index 1fa0acd..6462969 100644 --- a/preprocess/preprocessor.py +++ b/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}") diff --git a/runners/strategy_generator.py b/runners/strategy_generator.py index a9872a2..cf66224 100644 --- a/runners/strategy_generator.py +++ b/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