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

Add option to change .ply output filename for Gaussian exports (#3467)

Browse Source 
Add option to configure .ply output filename for Gaussian exports
This commit is contained in:
Brent Yi 2024-10-09 15:22:52 -07:00 committed by GitHub
parent 8e27e5f54d
commit 67be06009e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nerfstudio/scripts/exporter.py
View File

@ -491,6 +491,8 @@ class ExportGaussianSplat(Exporter):
Export 3D Gaussian Splatting model to a .ply
"""
output_filename: str = "splat.ply"
"""Name of the output file."""
obb_center: Optional[Tuple[float, float, float]] = None
"""Center of the oriented bounding box."""
obb_rotation: Optional[Tuple[float, float, float]] = None
@ -565,7 +567,7 @@ class ExportGaussianSplat(Exporter):
model: SplatfactoModel = pipeline.model
filename = self.output_dir / "splat.ply"
filename = self.output_dir / self.output_filename
map_to_tensors = OrderedDict()