Reverted include_background=True in Dice

C-Achard · C-Achard · commit b4b86f8574a6 · 2023-09-21T15:46:19.000+02:00
diff --git a/napari_cellseg3d/code_models/worker_training.py b/napari_cellseg3d/code_models/worker_training.py
@@ -154,7 +154,7 @@ def __init__(
         self.config = worker_config
 
         self.dice_metric = DiceMetric(
-            include_background=True, reduction="mean", get_not_nans=False
+            include_background=False, reduction="mean", get_not_nans=False
         )
         self.normalize_function = utils.remap_image
         self.start_time = time.time()
@@ -1309,7 +1309,7 @@ def get_patch_loader_func(num_samples):
                 else provided_scheduler
             )
             dice_metric = DiceMetric(
-                include_background=True, reduction="mean", ignore_empty=False
+                include_background=False, reduction="mean", ignore_empty=False
             )
 
             best_metric = -1

Original file line number	Diff line number	Diff line change
`@@ -154,7 +154,7 @@ def __init__(`
`154`	`154`	`self.config = worker_config`
`155`	`155`
`156`	`156`	`self.dice_metric = DiceMetric(`
`157`		`- include_background=True, reduction="mean", get_not_nans=False`
	`157`	`+ include_background=False, reduction="mean", get_not_nans=False`
`158`	`158`	`)`
`159`	`159`	`self.normalize_function = utils.remap_image`
`160`	`160`	`self.start_time = time.time()`
`@@ -1309,7 +1309,7 @@ def get_patch_loader_func(num_samples):`
`1309`	`1309`	`else provided_scheduler`
`1310`	`1310`	`)`
`1311`	`1311`	`dice_metric = DiceMetric(`
`1312`		`- include_background=True, reduction="mean", ignore_empty=False`
	`1312`	`+ include_background=False, reduction="mean", ignore_empty=False`
`1313`	`1313`	`)`
`1314`	`1314`
`1315`	`1315`	`best_metric = -1`