tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-12 10:55:57 +08:00
Code Activity

Merge branch 'main' into maturk/splatfacto_self_properties

Browse Source
This commit is contained in:
Matias Turkulainen authored and GitHub committed 2024-03-08 12:05:35 +02:00
2 parents 90af96c61f 6e6cedd438
commit 608b152fa5
2 files changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

nerfstudio/models/instant_ngp.py
+1 -1
View File
@@ -223,7 +223,7 @@ class NGPModel(Model):
return metrics_dict
def get_loss_dict(self, outputs, batch, metrics_dict=None):
image = batch["image"][..., :3].to(self.device)
image = batch["image"].to(self.device)
pred_rgb, image = self.renderer_rgb.blend_background_for_loss_computation(
pred_image=outputs["rgb"],
pred_accumulation=outputs["accumulation"],
nerfstudio/models/tensorf.py
+1 -1
View File
@@ -308,7 +308,7 @@ class TensoRFModel(Model):
def get_loss_dict(self, outputs, batch, metrics_dict=None) -> Dict[str, torch.Tensor]:
# Scaling metrics by coefficients to create the losses.
device = outputs["rgb"].device
image = batch["image"][..., :3].to(device)
image = batch["image"].to(device)
pred_image, image = self.renderer_rgb.blend_background_for_loss_computation(
pred_image=outputs["rgb"],
pred_accumulation=outputs["accumulation"],