mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-12 19:04:03 +08:00
Merge remote-tracking branch 'origin' into generative
This commit is contained in:
commit
c24ac22e61
2 files changed
+2
-2
No files matched your search
@@ -291,7 +291,7 @@ def depth_loss(
|
||||
Depth loss scalar.
|
||||
"""
|
||||
if not is_euclidean:
|
||||
termination_depth = termination_depth / directions_norm
|
||||
termination_depth = termination_depth * directions_norm
|
||||
steps = (ray_samples.frustums.starts + ray_samples.frustums.ends) / 2
|
||||
|
||||
if depth_loss_type == DephtLossType.DS_NERF:
|
||||
|
||||
@@ -108,7 +108,7 @@ class DepthNerfactoModel(NerfactoModel):
|
||||
metrics, images = super().get_image_metrics_and_images(outputs, batch)
|
||||
ground_truth_depth = batch["depth_image"]
|
||||
if not self.config.is_euclidean_depth:
|
||||
ground_truth_depth = ground_truth_depth / outputs["directions_norm"]
|
||||
ground_truth_depth = ground_truth_depth * outputs["directions_norm"]
|
||||
|
||||
ground_truth_depth_colormap = colormaps.apply_depth_colormap(ground_truth_depth)
|
||||
predicted_depth_colormap = colormaps.apply_depth_colormap(
|
||||
|
||||
Reference in new issue
Block a user