From 7ee48a43b64ed402c6d9b9d336db0f500246a21a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 6 Apr 2019 16:13:35 +0200 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 2b4358d2..351fccd2 100755 --- a/models.py +++ b/models.py @@ -159,7 +159,7 @@ class YOLOLayer(nn.Module): io = p.clone() # inference output io[..., 0:2] = torch.sigmoid(io[..., 0:2]) + self.grid_xy # xy io[..., 2:4] = torch.exp(io[..., 2:4]) * self.anchor_wh # wh yolo method - # io[..., 2:4] = ((torch.sigmoid(io[..., 2:4]) * 2) ** 2) * self.anchor_wh # wh power method + # io[..., 2:4] = ((torch.sigmoid(io[..., 2:4]) * 2) ** 3) * self.anchor_wh # wh power method io[..., 4:] = torch.sigmoid(io[..., 4:]) # p_conf, p_cls # io[..., 5:] = F.softmax(io[..., 5:], dim=4) # p_cls io[..., :4] *= self.stride