Config + disable remap in inference

Author: C-Achard

napari_cellseg3d/code_models/worker_inference.py

@@ -34,7 +34,6 @@
     ONNXModelWrapper,
     QuantileNormalization,
     QuantileNormalizationd,
-    RemapTensor,
     WeightsDownloader,
 )
 
@@ -718,15 +717,15 @@ def inference(self):
             if not post_process_config.thresholding.enabled:
                 post_process_transforms = Compose(
                     [
-                        RemapTensor(new_max=1.0, new_min=0.0),
+                        # RemapTensor(new_max=1.0, new_min=0.0),
                         EnsureType(),
                     ]
                 )
             else:
                 t = post_process_config.thresholding.threshold_value
                 post_process_transforms = Compose(
                     [
-                        RemapTensor(new_max=1.0, new_min=0.0),
+                        # RemapTensor(new_max=1.0, new_min=0.0),
                         AsDiscrete(threshold=t),
                         EnsureType(),
                     ]

napari_cellseg3d/code_plugins/plugin_model_training.py

@@ -186,7 +186,7 @@ def __init__(
 
         self.epoch_choice = ui.IntIncrementCounter(
             lower=2,
-            upper=200,
+            upper=999,
             default=self.default_config.max_epochs,
             text_label="Number of epochs : ",
         )

napari_cellseg3d/config.py

@@ -247,7 +247,7 @@ class TrainingWorkerConfig:
     weights_info: WeightsInfo = None
     train_data_dict: dict = None
     validation_percent: float = 0.8
-    max_epochs: int = 5
+    max_epochs: int = 50
     loss_function: callable = None
     learning_rate: np.float64 = 1e-3
     scheduler_patience: int = 10