Skip to content

Commit 960500c

Browse files
committed
fixed mask positions when using ff prediction
1 parent f751c21 commit 960500c

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

featureforest/make_zarr_dataset.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -110,6 +110,7 @@ def _(
110110
mask_browser,
111111
mo,
112112
np,
113+
patch_positions,
113114
):
114115
if not mask_browser.value:
115116
mo.stop(True)
@@ -126,7 +127,10 @@ def _(
126127
lazy_imread, mask_files, mask_shape, mask_dtype
127128
)
128129

129-
mask_positions = [get_patch_position(mfile) for mfile in mask_files]
130+
if "row_" in mask_files[0].stem:
131+
mask_positions = [get_patch_position(mfile) for mfile in mask_files]
132+
else:
133+
mask_positions = patch_positions
130134

131135
dask_mask = get_dask_image(
132136
lazy_masks, mask_positions, mask_shape, mask_dtype

0 commit comments

Comments
 (0)