diff --git a/configs/local/view_generate_config.yaml b/configs/local/view_generate_config.yaml index c9a5234..7364d5d 100644 --- a/configs/local/view_generate_config.yaml +++ b/configs/local/view_generate_config.yaml @@ -7,9 +7,9 @@ runner: name: debug root_dir: experiments generate: - port: 5002 - from: 2200 - to: 2300 # -1 means all + port: 5004 + from: 4000 + to: -1 # -1 means all object_dir: /media/hofee/data/data/scaled_object_meshes table_model_path: /media/hofee/data/data/others/table.obj output_dir: /media/hofee/repository/new_data_with_normal @@ -49,4 +49,4 @@ runner: Light: location: [0,0,3.5] orientation: [0,0,0] - power: 150 + power: 150 \ No newline at end of file diff --git a/preprocess/preprocessor.py b/preprocess/preprocessor.py index e7fd616..85f4460 100644 --- a/preprocess/preprocessor.py +++ b/preprocess/preprocessor.py @@ -78,7 +78,11 @@ def save_scene_data(root, scene, scene_idx=0, scene_total=1,file_type="txt"): ''' read frame data(depth|mask|normal) ''' frame_num = DataLoadUtil.get_scene_seq_length(root, scene) for frame_id in range(frame_num): + + print(f"[scene({scene_idx}/{scene_total})|frame({frame_id}/{frame_num})]Processing {scene} frame {frame_id}") + if frame_id != 126: + continue path = DataLoadUtil.get_path(root, scene, frame_id) cam_info = DataLoadUtil.load_cam_info(path, binocular=True) depth_L, depth_R = DataLoadUtil.load_depth( @@ -148,6 +152,8 @@ if __name__ == "__main__": from_idx = 0 to_idx = len(scene_list) + + cnt = 0 import time total = to_idx - from_idx @@ -156,4 +162,4 @@ if __name__ == "__main__": save_scene_data(root, scene, cnt, total, file_type="npy") cnt+=1 end = time.time() - print(f"Time cost: {end-start}") \ No newline at end of file + print(f"Time cost: {end-start}")