diff --git a/preprocess/preprocessor.py b/preprocess/preprocessor.py index 0fccce1..08f7dfd 100644 --- a/preprocess/preprocessor.py +++ b/preprocess/preprocessor.py @@ -23,9 +23,9 @@ def save_target_points(root, scene, frame_idx, target_points: np.ndarray, file_t save_np_pts(pts_path, target_points, file_type) def save_target_normals(root, scene, frame_idx, target_normals: np.ndarray, file_type="txt"): - pts_path = os.path.join(root,scene, "normals", f"{frame_idx}.{file_type}") - if not os.path.exists(os.path.join(root,scene, "normals")): - os.makedirs(os.path.join(root,scene, "normals")) + pts_path = os.path.join(root,scene, "nrm", f"{frame_idx}.{file_type}") + if not os.path.exists(os.path.join(root,scene, "nrm")): + os.makedirs(os.path.join(root,scene, "nrm")) save_np_pts(pts_path, target_normals, file_type) def save_scan_points_indices(root, scene, frame_idx, scan_points_indices: np.ndarray, file_type="txt"):