workaround naver#15

abecadel · Jul 17, 2024 · 1253bce · 1253bce
1 parent 6d6d485
commit 1253bce
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mast3r/cloud_opt/sparse_ga.py b/mast3r/cloud_opt/sparse_ga.py
@@ -823,7 +823,8 @@ def canonical_view(ptmaps11, confs11, subsample, mode='avg-angle'):
     canon_depth = ptmaps11[..., 2].unsqueeze(1)
     S = slice(subsample // 2, None, subsample)
     center_depth = canon_depth[:, :, S, S]
-    assert (center_depth > 0).all()
+    center_depth = torch.clip(center_depth, min=torch.finfo(center_depth.dtype).eps)
+
     stacked_depth = F.pixel_unshuffle(canon_depth, subsample)
     stacked_confs = F.pixel_unshuffle(confs11[:, None, :, :, 0], subsample)