From 29ddd2ca3a33c43fb8d44b90d2758aaba0436f57 Mon Sep 17 00:00:00 2001 From: Brent Yi Date: Thu, 22 Feb 2024 19:26:59 -0800 Subject: [PATCH] Support new MPS filenames for Project Aria point clouds (#2949) * Support new point cloud filename for Project Aria point clouds * Comment details --- nerfstudio/scripts/datasets/process_project_aria.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/nerfstudio/scripts/datasets/process_project_aria.py b/nerfstudio/scripts/datasets/process_project_aria.py index 23d304cf6..fe4874832 100644 --- a/nerfstudio/scripts/datasets/process_project_aria.py +++ b/nerfstudio/scripts/datasets/process_project_aria.py @@ -223,6 +223,11 @@ class ProcessProjectAria: # save global point cloud, which is useful for Gaussian Splatting. points_path = self.mps_data_dir / "global_points.csv.gz" + if not points_path.exists(): + # MPS point cloud output was renamed in Aria's December 4th, 2023 update. + # https://facebookresearch.github.io/projectaria_tools/docs/ARK/sw_release_notes#project-aria-updates-aria-mobile-app-v140-and-changes-to-mps + points_path = self.mps_data_dir / "semidense_points.csv.gz" + if points_path.exists(): print("Found global points, saving to PLY...") points_data = mps.read_global_point_cloud(str(points_path)) # type: ignore