diff --git a/inference.py b/inference.py index 7b846d8..2547133 100644 --- a/inference.py +++ b/inference.py @@ -101,7 +101,7 @@ def main(args): i_s = torch.sum(intersection) p_s = torch.sum(pred_choice) t_s = torch.sum(seg) - dice = 1- (2*(i_s)+1)/(p_s+t_s+1) + dice = (2*(i_s)+1)/(p_s+t_s+1) ave_dice=torch.add(ave_dice,dice) ####################################