From f1a7d5383a5a3ccaea3a6a266c33fe1898a0ef6d Mon Sep 17 00:00:00 2001 From: Treeboy2762 <44747571+Treeboy2762@users.noreply.github.com> Date: Mon, 18 Jul 2022 11:41:38 +0900 Subject: [PATCH] Update lovasz_losses.py 1-gt.sorted to (~(gt.sorted)) --- src/criterions/lovasz_losses.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/criterions/lovasz_losses.py b/src/criterions/lovasz_losses.py index d83f292..a090f11 100644 --- a/src/criterions/lovasz_losses.py +++ b/src/criterions/lovasz_losses.py @@ -23,7 +23,7 @@ def lovasz_grad(gt_sorted): p = len(gt_sorted) gts = gt_sorted.sum() intersection = gts.float() - gt_sorted.float().cumsum(0) - union = gts.float() + (1 - gt_sorted).float().cumsum(0) + union = gts.float() + (~(gt_sorted)).float().cumsum(0) jaccard = 1. - intersection / union if p > 1: # cover 1-pixel case jaccard[1:p] = jaccard[1:p] - jaccard[0:-1]