From 6340074c2a1890100007cf66cb71e6a07b1531b8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 5 Dec 2019 20:24:42 -0800 Subject: [PATCH] updates --- utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/utils.py b/utils/utils.py index fe143fdb..0c2c2066 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -359,7 +359,7 @@ def compute_loss(p, targets, model): # predictions, targets, model # GIoU pxy = torch.sigmoid(ps[:, 0:2]) # pxy = pxy * s - (s - 1) / 2, s = 1.5 (scale_xy) - pbox = torch.cat((pxy, torch.exp(ps[:, 2:4]).clamp(max=1E4) * anchor_vec[i]), 1) # predicted box + pbox = torch.cat((pxy, torch.exp(ps[:, 2:4]).clamp(max=1E3) * anchor_vec[i]), 1) # predicted box giou = bbox_iou(pbox.t(), tbox[i], x1y1x2y2=False, GIoU=True) # giou computation lbox += (1.0 - giou).mean() # giou loss