AcquisitionFunction.compute_acquisition
AcquisitionFunction.compute_acquisition(
dataset_metadata,
output_axes,
mask_axes,
reference_scale,
acquisition_fun,
segmentation_out,
sampled_mask=None,
segmentation_only=False,
)