diff --git a/eval.py b/eval.py index 1aada5e..c727510 100644 --- a/eval.py +++ b/eval.py @@ -23,4 +23,4 @@ def eval_net(net, dataset, gpu=False): mask_pred = (mask_pred > 0.5).float() tot += dice_coeff(mask_pred, true_mask).item() - return tot / i + return tot / (i + 1)