Update worker_training.py

C-Achard · C-Achard · commit 4490b019686b · 2023-07-19T12:23:57.000+02:00
diff --git a/napari_cellseg3d/code_models/worker_training.py b/napari_cellseg3d/code_models/worker_training.py
@@ -44,6 +44,7 @@
     PRETRAINED_WEIGHTS_DIR,
     LogSignal,
     QuantileNormalizationd,
+    RemapTensor,
     TrainingReport,
     WeightsDownloader,
 )
@@ -639,7 +640,7 @@ def get_loader_func(num_samples):
                             # TODO : more parameters/flexibility
                             post_pred = Compose(
                                 [
-                                    # RemapTensor(new_max=1, new_min=0),
+                                    RemapTensor(new_max=1, new_min=0),
                                     AsDiscrete(threshold=0.5),  # needed ?
                                     EnsureType(),
                                 ]

Original file line number	Diff line number	Diff line change
`@@ -44,6 +44,7 @@`
`44`	`44`	`PRETRAINED_WEIGHTS_DIR,`
`45`	`45`	`LogSignal,`
`46`	`46`	`QuantileNormalizationd,`
	`47`	`+ RemapTensor,`
`47`	`48`	`TrainingReport,`
`48`	`49`	`WeightsDownloader,`
`49`	`50`	`)`
`@@ -639,7 +640,7 @@ def get_loader_func(num_samples):`
`639`	`640`	`# TODO : more parameters/flexibility`
`640`	`641`	`post_pred = Compose(`
`641`	`642`	`[`
`642`		`- # RemapTensor(new_max=1, new_min=0),`
	`643`	`+ RemapTensor(new_max=1, new_min=0),`
`643`	`644`	`AsDiscrete(threshold=0.5), # needed ?`
`644`	`645`	`EnsureType(),`
`645`	`646`	`]`