finish pipeline

utils/preprocess_util.py

@@ -8,6 +8,7 @@ sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 from utils.reconstruction_util import ReconstructionUtil
 from utils.data_load import DataLoadUtil
 from utils.pts_util import PtsUtil
+from PytorchBoot.utils.log_util import Log
 
 def save_np_pts(path, pts: np.ndarray, file_type="txt"):
     if file_type == "txt":
@@ -75,7 +76,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}")
+        Log.info(f"[frame({frame_id}/{frame_num})]Processing {scene} frame {frame_id}")
         path = DataLoadUtil.get_path(root, scene, frame_id)
         cam_info = DataLoadUtil.load_cam_info(path, binocular=True)
         depth_L, depth_R = DataLoadUtil.load_depth(