tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 10:34:29 +08:00
Code

Support new MPS filenames for Project Aria point clouds (#2949)

Browse Source 
* Support new point cloud filename for Project Aria point clouds

* Comment details
This commit is contained in:
Brent Yi 2024-02-22 19:26:59 -08:00 committed by GitHub
parent cc0205943a
commit 29ddd2ca3a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nerfstudio/scripts/datasets/process_project_aria.py
View File

@ -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