From 00cc02ba915b79e9fe8a200f880a7d5bee78c998 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 15 Apr 2019 13:58:48 +0200 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 7a584026..4dd3c10f 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -265,7 +265,7 @@ def compute_loss(p, targets): # predictions, targets lxy += (k * 8) * MSE(torch.sigmoid(pi[..., 0:2]), txy[i]) # xy loss lwh += (k * 1) * MSE(pi[..., 2:4], twh[i]) # wh yolo loss - # lwh += (k * 4) * MSE(torch.sigmoid(pi[..., 2:4]), twh[i]) # wh power loss + # lwh += (k * 1) * MSE(torch.sigmoid(pi[..., 2:4]), twh[i]) # wh power loss lcls += (k * 1) * CE(pi[..., 5:], tcls[i]) # class_conf loss # pos_weight = FT([gp[i] / min(gp) * 4.])