Merge pull request #45 from dmitrysarov/master

evaluation fix : index + 1 in average Former-commit-id: c8179b82ebe715817890a967b6775ef7875007b3
2019-01-09 14:32:55 +01:00 · 2019-01-09 14:32:55 +01:00 · 0f5e7cb0f9
parent ef36cad9ba 078b03ffa9
commit 0f5e7cb0f9
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)