diff --git a/configs/local/strategy_generate_config.yaml b/configs/local/strategy_generate_config.yaml index 3e4576d..88d1733 100644 --- a/configs/local/strategy_generate_config.yaml +++ b/configs/local/strategy_generate_config.yaml @@ -28,8 +28,8 @@ runner: datasets: OmniObject3d: #"/media/hofee/data/data/temp_output" - root_dir: "/media/hofee/repository/new_full_box_data" - model_dir: "/media/hofee/data/data/scaled_object_meshes" + root_dir: "/media/hofee/data/tempdir/test_real_output" + model_dir: "/media/hofee/data/tempdir/test_real" from: 0 to: -1 # -1 means end #output_dir: "/media/hofee/data/data/label_output" diff --git a/configs/local/view_generate_config.yaml b/configs/local/view_generate_config.yaml index 7364d5d..fbfaeca 100644 --- a/configs/local/view_generate_config.yaml +++ b/configs/local/view_generate_config.yaml @@ -8,11 +8,11 @@ runner: root_dir: experiments generate: port: 5004 - from: 4000 - to: -1 # -1 means all - object_dir: /media/hofee/data/data/scaled_object_meshes + from: 0 + to: 1 # -1 means all + object_dir: /media/hofee/data/tempdir/test_real table_model_path: /media/hofee/data/data/others/table.obj - output_dir: /media/hofee/repository/new_data_with_normal + output_dir: /media/hofee/data/tempdir/test_real_output binocular_vision: true plane_size: 10 max_views: 512 @@ -20,13 +20,13 @@ runner: random_view_ratio: 0.2 min_cam_table_included_degree: 20 max_diag: 0.7 - min_diag: 0.1 + min_diag: 0.01 random_config: display_table: min_height: 0.05 max_height: 0.15 - min_radius: 0.3 - max_radius: 0.5 + min_radius: 0.2 + max_radius: 0.3 display_object: min_x: 0 max_x: 0.03 @@ -44,7 +44,7 @@ runner: far_plane: 5 fov_vertical: 25 resolution: [1280,800] - eye_distance: 0.15 + eye_distance: 0.10 eye_angle: 25 Light: location: [0,0,3.5] diff --git a/preprocess/preprocessor.py b/preprocess/preprocessor.py index dd48ee4..1eb56b2 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"): voxel_size=0.002 filter_degree = 75 min_z = 0.2 - max_z = 0.45 + max_z = 0.5 ''' scan points ''' display_table_info = DataLoadUtil.get_display_table_info(root, scene) @@ -138,16 +138,16 @@ 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/repository/new_data_with_normal" - list_path = r"/media/hofee/repository/full_list.txt" - scene_list = [] + root = r"/media/hofee/data/tempdir/test_real_output" + # list_path = r"/media/hofee/repository/full_list.txt" + # scene_list = [] - with open(list_path, "r") as f: - for line in f: - scene_list.append(line.strip()) - - from_idx = 1000 - to_idx = 1500 + # with open(list_path, "r") as f: + # for line in f: + # scene_list.append(line.strip()) + scene_list = os.listdir(root) + from_idx = 0 # 1000 + to_idx = 1 # 1500 cnt = 0 @@ -155,7 +155,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, "npy") + save_scene_data(root, scene, cnt, total, "txt") cnt+=1 end = time.time() print(f"Time cost: {end-start}")