Update loss.py

ultralytics · Nov 25, 2020 · 93120f2 · 93120f2
1 parent 25239e1
commit 93120f2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/utils/loss.py b/utils/loss.py
@@ -70,9 +70,7 @@ def __init__(self, loss_fcn, gamma=1.5, alpha=0.25):
         self.loss_fcn.reduction = 'none'  # required to apply FL to each element
 
     def forward(self, pred, true):
-        p_loss = self.loss_fcn(pred, true)
-        n_loss = self.loss_fcn(-pred, 1 - true)
-        loss = (true) * p_loss + (1 - true) * n_loss
+        loss = self.loss_fcn(pred, true)
 
         pred_prob = torch.sigmoid(pred)  # prob from logits
         alpha_factor = true * self.alpha + (1 - true) * (1 - self.alpha)