|
15 | 15 | from monai.deploy.operators.monai_bundle_inference_operator import MonaiBundleInferenceOperator, get_bundle_config |
16 | 16 | from monai.deploy.utils.importutil import optional_import |
17 | 17 |
|
18 | | -MONAI_UTILS = "monai.utils" |
19 | | -nibabel, _ = optional_import("nibabel", "3.2.1") |
20 | 18 | torch, _ = optional_import("torch", "1.10.2") |
21 | 19 |
|
22 | | -NdarrayOrTensor, _ = optional_import("monai.config", name="NdarrayOrTensor") |
23 | | -MetaTensor, _ = optional_import("monai.data.meta_tensor", name="MetaTensor") |
24 | | -PostFix, _ = optional_import("monai.utils.enums", name="PostFix") # For the default meta_key_postfix |
25 | | -first, _ = optional_import("monai.utils.misc", name="first") |
26 | | -ensure_tuple, _ = optional_import(MONAI_UTILS, name="ensure_tuple") |
27 | | -convert_to_dst_type, _ = optional_import(MONAI_UTILS, name="convert_to_dst_type") |
28 | | -Key, _ = optional_import(MONAI_UTILS, name="ImageMetaKey") |
29 | | -MetaKeys, _ = optional_import(MONAI_UTILS, name="MetaKeys") |
30 | | -SpaceKeys, _ = optional_import(MONAI_UTILS, name="SpaceKeys") |
31 | | -Compose_, _ = optional_import("monai.transforms", name="Compose") |
32 | | -ConfigParser_, _ = optional_import("monai.bundle", name="ConfigParser") |
33 | | -MapTransform_, _ = optional_import("monai.transforms", name="MapTransform") |
34 | | -SimpleInferer, _ = optional_import("monai.inferers", name="SimpleInferer") |
35 | | - |
36 | | -Compose: Any = Compose_ |
37 | | -MapTransform: Any = MapTransform_ |
38 | | -ConfigParser: Any = ConfigParser_ |
39 | 20 | __all__ = ["MONetBundleInferenceOperator"] |
40 | 21 |
|
41 | 22 |
|
@@ -82,7 +63,7 @@ def predict(self, data: Any, *args, **kwargs) -> Union[Image, Any, Tuple[Any, .. |
82 | 63 | """Predicts output using the inferer.""" |
83 | 64 |
|
84 | 65 | self._nnunet_predictor.predictor.network = self._model_network |
85 | | - # os.environ['nnUNet_def_n_proc'] = "1" |
| 66 | + |
86 | 67 | if len(data.shape) == 4: |
87 | 68 | data = data[None] |
88 | 69 | return self._nnunet_predictor(data) |
0 commit comments