diff --git a/monai/auto3dseg/utils.py b/monai/auto3dseg/utils.py index 7aee846225..58b900d410 100644 --- a/monai/auto3dseg/utils.py +++ b/monai/auto3dseg/utils.py @@ -94,11 +94,11 @@ def get_label_ccp(mask_index: MetaTensor, use_gpu: bool = True) -> tuple[list[An regardless of this setting. """ - cucim, has_cucim = optional_import("cucim") + skimage, has_cucim = optional_import("cucim.skimage") shape_list = [] if mask_index.device.type == "cuda" and has_cp and has_cucim and use_gpu: mask_cupy = ToCupy()(mask_index.short()) - labeled = cucim.skimage.measure.label(mask_cupy) + labeled = skimage.measure.label(mask_cupy) vals = cp.unique(labeled[cp.nonzero(labeled)]) for ncomp in vals: diff --git a/monai/transforms/utils.py b/monai/transforms/utils.py index 33d468b86e..b769b8da2f 100644 --- a/monai/transforms/utils.py +++ b/monai/transforms/utils.py @@ -1027,11 +1027,11 @@ def get_largest_connected_component_mask( """ # use skimage/cucim.skimage and np/cp depending on whether packages are # available and input is non-cpu torch.tensor - cucim, has_cucim = optional_import("cucim") + skimage, has_cucim = optional_import("cucim.skimage") use_cp = has_cp and has_cucim and isinstance(img, torch.Tensor) and img.device != torch.device("cpu") if use_cp: img_ = convert_to_cupy(img.short()) # type: ignore - label = cucim.skimage.measure.label + label = skimage.measure.label lib = cp else: if not has_measure: