Skip to content

Commit c841ce1

Browse files
committed
Add option to disable world space normalization in simple_trainer_2dgs.py
1 parent 0b4dddf commit c841ce1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

examples/simple_trainer_2dgs.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,8 @@ class Config:
5454
patch_size: Optional[int] = None
5555
# A global scaler that applies to the scene size related parameters
5656
global_scale: float = 1.0
57+
# Normalize the world space
58+
normalize_world_space: bool = True
5759

5860
# Port for the viewer server
5961
port: int = 8080
@@ -275,7 +277,7 @@ def __init__(self, cfg: Config) -> None:
275277
self.parser = Parser(
276278
data_dir=cfg.data_dir,
277279
factor=cfg.data_factor,
278-
normalize=True,
280+
normalize=cfg.normalize_world_space,
279281
test_every=cfg.test_every,
280282
)
281283
self.trainset = Dataset(

0 commit comments

Comments
 (0)