diff --git a/preprocess/preprocessor.py b/preprocess/preprocessor.py index 85f4460..c95a3b2 100644 --- a/preprocess/preprocessor.py +++ b/preprocess/preprocessor.py @@ -78,11 +78,7 @@ 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( @@ -142,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"C:\Document\Local Project\nbv_rec\nbv_reconstruction\test\test_sample" - list_path = r"C:\Document\Local Project\nbv_rec\nbv_reconstruction\test\test_sample/test_sample_list.txt" + root = r"/media/hofee/repository/new_data_with_normal" + 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 = 0 - to_idx = len(scene_list) + from_idx = 1000 + to_idx = 1500 cnt = 0