Fix network loading

monailabel/tasks/infer/bundle.py

@@ -87,6 +87,7 @@ def __init__(
         extend_load_image: bool = True,
         add_post_restore: bool = True,
         dropout: float = 0.0,
+        load_strict=False,
         **kwargs,
     ):
         self.valid: bool = False
@@ -149,6 +150,7 @@ def __init__(
             dimension=dimension,
             description=description,
             preload=strtobool(conf.get("preload", "false")),
+            load_strict=load_strict,
             **kwargs,
         )
 

sample-apps/endoscopy/lib/infers/deepedit.py

@@ -55,6 +55,7 @@ def __init__(
             labels=labels,
             dimension=dimension,
             description=description,
+            load_strict=False,
             **kwargs,
         )
 

sample-apps/endoscopy/lib/infers/inbody.py

@@ -24,7 +24,7 @@ class InBody(BundleInferTask):
     """
 
     def __init__(self, path: str, conf: Dict[str, str], **kwargs):
-        super().__init__(path, conf, type=InferType.CLASSIFICATION, add_post_restore=False, **kwargs)
+        super().__init__(path, conf, type=InferType.CLASSIFICATION, add_post_restore=False, load_strict=False, **kwargs)
 
         # Override Labels
         self.labels = {"InBody": 0, "OutBody": 1}

sample-apps/endoscopy/lib/infers/tooltracking.py

@@ -28,7 +28,7 @@ class ToolTracking(BundleInferTask):
     """
 
     def __init__(self, path: str, conf: Dict[str, str], **kwargs):
-        super().__init__(path, conf, type=InferType.SEGMENTATION, **kwargs)
+        super().__init__(path, conf, type=InferType.SEGMENTATION, load_strict=False, **kwargs)
 
         # Override Labels
         self.labels = {"Tool": 1}

sample-apps/pathology/lib/infers/classification_nuclei.py

@@ -24,7 +24,7 @@ class ClassificationNuclei(BundleInferTask):
     """
 
     def __init__(self, path: str, conf: Dict[str, str], **kwargs):
-        super().__init__(path, conf, type=InferType.CLASSIFICATION, add_post_restore=False, **kwargs)
+        super().__init__(path, conf, type=InferType.CLASSIFICATION, add_post_restore=False, load_strict=False, **kwargs)
 
         # Override Labels
         self.labels = {

sample-apps/pathology/lib/infers/nuclick.py

@@ -38,6 +38,7 @@ def __init__(self, path: str, conf: Dict[str, str], **kwargs):
             conf,
             type=InferType.ANNOTATION,
             add_post_restore=False,
+            load_strict=False,
             **kwargs,
             pre_filter=[LoadImaged, SqueezeDimd],
             post_filter=[KeepLargestConnectedComponentd, SaveImaged],

sample-apps/pathology/lib/infers/segmentation_nuclei.py

@@ -48,6 +48,7 @@ def __init__(
             labels=labels,
             dimension=dimension,
             description=description,
+            load_strict=False,
             **kwargs,
         )
 

sample-apps/radiology/lib/infers/deepedit.py

@@ -65,6 +65,7 @@ def __init__(
             input_key="image",
             output_label_key="pred",
             output_json_key="result",
+            load_strict=False,
             **kwargs,
         )
 

sample-apps/radiology/lib/infers/deepgrow.py

@@ -63,6 +63,7 @@ def __init__(
             labels=labels,
             dimension=dimension,
             description=description,
+            load_strict=False,
             **kwargs,
         )
 

sample-apps/radiology/lib/infers/deepgrow_pipeline.py

@@ -68,6 +68,7 @@ def __init__(
             dimension=dimension,
             description=description,
             config={"cache_transforms": True, "cache_transforms_in_memory": True, "cache_transforms_ttl": 300},
+            load_strict=False,
         )
         self.model_3d = model_3d
         self.spatial_size = spatial_size

sample-apps/radiology/lib/infers/localization_spine.py

@@ -54,6 +54,7 @@ def __init__(
             labels=labels,
             dimension=dimension,
             description=description,
+            load_strict=False,
             **kwargs,
         )
         self.target_spacing = target_spacing

sample-apps/radiology/lib/infers/localization_vertebra.py

@@ -56,6 +56,7 @@ def __init__(
             labels=labels,
             dimension=dimension,
             description=description,
+            load_strict=False,
             **kwargs,
         )
         self.target_spacing = target_spacing

sample-apps/radiology/lib/infers/segmentation.py

@@ -54,6 +54,7 @@ def __init__(
             labels=labels,
             dimension=dimension,
             description=description,
+            load_strict=False,
             **kwargs,
         )
         self.target_spacing = target_spacing

sample-apps/radiology/lib/infers/segmentation_spleen.py

@@ -53,6 +53,7 @@ def __init__(
             labels=labels,
             dimension=dimension,
             description=description,
+            load_strict=False,
             **kwargs,
         )
         self.target_spacing = target_spacing

sample-apps/radiology/lib/infers/segmentation_vertebra.py

@@ -63,6 +63,7 @@ def __init__(
             labels=labels,
             dimension=dimension,
             description=description,
+            load_strict=False,
             **kwargs,
         )
         self.target_spacing = target_spacing

sample-apps/radiology/lib/infers/vertebra_pipeline.py

@@ -47,6 +47,7 @@ def __init__(
             labels=task_seg_vertebra.labels,
             dimension=task_seg_vertebra.dimension,
             description=description,
+            load_strict=False,
             **kwargs,
         )