Skip to content

Commit 3ecbcd7

Browse files
committed
minor depth ensembling fixes
1 parent b3d1152 commit 3ecbcd7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/diffusers/pipelines/marigold/pipeline_marigold_depth.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -736,6 +736,7 @@ def init_param(depth: torch.Tensor):
736736
param = init_s.cpu().numpy()
737737
else:
738738
raise ValueError("Unrecognized alignment.")
739+
param = param.astype(np.float64)
739740

740741
return param
741742

@@ -778,7 +779,7 @@ def cost_fn(param: np.ndarray, depth: torch.Tensor) -> float:
778779

779780
if regularizer_strength > 0:
780781
prediction, _ = ensemble(depth_aligned, return_uncertainty=False)
781-
err_near = (0.0 - prediction.min()).abs().item()
782+
err_near = prediction.min().abs().item()
782783
err_far = (1.0 - prediction.max()).abs().item()
783784
cost += (err_near + err_far) * regularizer_strength
784785

0 commit comments

Comments
 (0)