Refactor nnUNet bundle inference operator to MONetBundleInferenceOperator and add implementation

Author: SimoneBendazzoli93

examples/apps/ai_spleen_nnunet_seg_app/app.py

@@ -27,7 +27,7 @@
     IOMapping,
 
 )
-from monai.deploy.operators.monai_nnunet_bundle_inference_operator import MonainnUNetBundleInferenceOperator
+from monai.deploy.operators.monai_nnunet_bundle_inference_operator import MONetBundleInferenceOperator
 from monai.deploy.operators.stl_conversion_operator import STLConversionOperator
 
 
@@ -87,7 +87,7 @@ def compose(self):
 
         config_names = BundleConfigNames(config_names=["inference"])  # Same as the default
 
-        bundle_spleen_seg_op = MonainnUNetBundleInferenceOperator(
+        bundle_spleen_seg_op = MONetBundleInferenceOperator(
             self,
             input_mapping=[IOMapping("image", Image, IOType.IN_MEMORY)],
             output_mapping=[IOMapping("pred", Image, IOType.IN_MEMORY)],

monai/deploy/operators/__init__.py

@@ -48,7 +48,7 @@
 from .dicom_utils import EquipmentInfo, ModelInfo, random_with_n_digits, save_dcm_file, write_common_modules
 from .inference_operator import InferenceOperator
 from .monai_bundle_inference_operator import BundleConfigNames, IOMapping, MonaiBundleInferenceOperator
-from .monai_nnunet_bundle_inference_operator import MonainnUNetBundleInferenceOperator
+from .monet_bundle_inference_operator import MONetBundleInferenceOperator
 from .monai_seg_inference_operator import MonaiSegInferenceOperator
 from .nii_data_loader_operator import NiftiDataLoader
 from .png_converter_operator import PNGConverterOperator

monai/deploy/operators/monai_nnunet_bundle_inference_operator.py renamed to monai/deploy/operators/monet_bundle_inference_operator.py

@@ -13,7 +13,7 @@
 from monai.deploy.utils.importutil import optional_import
 from typing import Any, Dict, Tuple, Union
 from monai.deploy.core import Image
-from pathlib import Path
+
 MONAI_UTILS = "monai.utils"
 nibabel, _ = optional_import("nibabel", "3.2.1")
 torch, _ = optional_import("torch", "1.10.2")
@@ -35,10 +35,10 @@
 Compose: Any = Compose_
 MapTransform: Any = MapTransform_
 ConfigParser: Any = ConfigParser_
-__all__ = ["MonainnUNetBundleInferenceOperator"]
+__all__ = ["MONetBundleInferenceOperator"]
 
 
-class MonainnUNetBundleInferenceOperator(MonaiBundleInferenceOperator):
+class MONetBundleInferenceOperator(MonaiBundleInferenceOperator):
     """
     A specialized operator for performing inference using the MONAI nnUNet bundle.
     This operator extends the `MonaiBundleInferenceOperator` to support nnUNet-specific