index + 1 in average

Former-commit-id: 1dc82b17f075a74f7ef6b20c888835497bcd690b
2024-09-19 11:35:28 +00:00 · 2019-01-09 15:01:42 +03:00 · 2019-01-09 15:01:42 +03:00 · 078b03ffa9
parent ef36cad9ba
commit 078b03ffa9
1 changed files with 1 additions and 1 deletions
--- 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)