feat(interpolate_flow): expose mask_value_thr

ZettaAI · Feb 17, 2025 · 1955e4f · 1955e4f
1 parent d79f9f6
commit 1955e4f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/zetta_utils/mazepa_layer_processing/common/interpolate_flow.py b/zetta_utils/mazepa_layer_processing/common/interpolate_flow.py
@@ -70,6 +70,7 @@ def build_interpolate_flow(  # pylint: disable=too-many-locals
     bbox: BBox3D | None = None,
     auto_bbox: bool = False,
     dst_tighten_bounds: bool = False,
+    mask_value_thr: float = 0,
 ) -> mazepa.Flow:
     if dst is None:
         dst = src
@@ -100,6 +101,7 @@ def build_interpolate_flow(  # pylint: disable=too-many-locals
                 op=make_interpolate_operation(
                     res_change_mult=dst_res / last_res,
                     mode=mode,
+                    mask_value_thr=mask_value_thr,
                 ),
                 dst=dst,
                 dst_resolution=dst_res,