Update loss.py

mkang315 · May 14, 2024 · 1bd43fe · 1bd43fe
1 parent 0829160
commit 1bd43fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/loss.py b/utils/loss.py
@@ -138,7 +138,7 @@ def __call__(self, p, targets):  # predictions, targets
                 pxy = pxy.sigmoid() * 2 - 0.5
                 pwh = (pwh.sigmoid() * 2) ** 2 * anchors[i]
                 pbox = torch.cat((pxy, pwh), 1)  # predicted box
-                iou = bbox_iou(pbox, tbox[i], ESIoU=True).squeeze()  # iou(prediction, target)
+                iou = bbox_iou(pbox, tbox[i], EIoU=True).squeeze()  # iou(prediction, target)
                 lbox += (1.0 - iou).mean()  # iou loss
 
                 # Objectness