From 67be06009eae045f25b1ce5a67c012cbaf57e2ab Mon Sep 17 00:00:00 2001 From: Brent Yi Date: Wed, 9 Oct 2024 15:22:52 -0700 Subject: [PATCH] Add option to change .ply output filename for Gaussian exports (#3467) Add option to configure .ply output filename for Gaussian exports --- nerfstudio/scripts/exporter.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/nerfstudio/scripts/exporter.py b/nerfstudio/scripts/exporter.py index fccb70ffb..9c0812e3e 100644 --- a/nerfstudio/scripts/exporter.py +++ b/nerfstudio/scripts/exporter.py @@ -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()