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

Add option to configure .ply output filename for Gaussian exports
This commit is contained in:
Brent Yi authored and GitHub committed 2024-10-09 22:22:52 +00:00
1 parent 8e27e5f54d
commit 67be06009e
1 file changed
+3 -1
+3 -1
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()