Merge remote-tracking branch 'origin' into generative

This commit is contained in:
Matt Tancik committed 2023-02-03 11:21:23 -08:00
commit c24ac22e61
2 files changed
+2 -2

No files matched your search

+1 -1
View File
@@ -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:
+1 -1
View File
@@ -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(