masks_to_flows_batch

omnipose.core.masks_to_flows_batch(batch, links=[None], device=device(type='cpu'), omni=True, dim=2, smooth=False, normalize=False, affinity_field=False, initialize=False, n_iter=None, verbose=False)

Batch process flows. This includes padding with relection to not have weird cutoff flows.

Parameters

mask_batch (list, NDarray) -- list of masks all of shape tyx

Return type

concatenated labels, links, etc. and slices to extract them