tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Code

params, remove normalization on quats

Browse Source
This commit is contained in:
Justin Kerr 2023-10-13 00:03:35 -07:00
parent 6e20b7b418
commit bef506475f
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nerfstudio/configs/method_configs.py
View File

@ -643,14 +643,14 @@ method_configs["gaussian-splatting"] = TrainerConfig(
"optimizer": AdamOptimizerConfig(lr=0.005, eps=1e-15),
"scheduler": ExponentialDecaySchedulerConfig(
lr_final=1e-3,
max_steps=3000,
max_steps=15000,
),
},
"rotation": {
"optimizer": AdamOptimizerConfig(lr=3e-4, eps=1e-15),
"optimizer": AdamOptimizerConfig(lr=.001, eps=1e-15),
"scheduler": ExponentialDecaySchedulerConfig(
lr_final=1e-4,
max_steps=3000,
max_steps=15000,
),
},
},

2
nerfstudio/models/gaussian_splatting.py
View File

@ -572,7 +572,7 @@ class GaussianSplattingModel(Model):
self.means[crop_ids],
torch.exp(self.scales[crop_ids]),
1,
self.quats[crop_ids]/self.quats[crop_ids].norm(dim=-1,keepdim=True),
self.quats[crop_ids],#/self.quats[crop_ids].norm(dim=-1,keepdim=True),
viewmat.squeeze()[:3, :],
projmat.squeeze() @ viewmat.squeeze(),
camera.fx.item(),