From 75225e4d990ffc1e9e944b2a497222ec781f979e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 19 Feb 2019 19:30:56 +0100 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index e83a68d1..23f48d00 100755 --- a/models.py +++ b/models.py @@ -194,7 +194,7 @@ class YOLOLayer(nn.Module): else: if ONNX_EXPORT: grid_xy = self.grid_xy.repeat((1, self.nA, 1, 1, 1)).view((1, -1, 2)) - anchor_wh = self.anchor_wh.repeat((1, 1, nG, nG, 1)).view((1, -1, 2)) + anchor_wh = self.anchor_wh.repeat((1, 1, nG, nG, 1)).view((1, -1, 2)) / nG # p = p.view(-1, 85) # xy = torch.sigmoid(p[:, 0:2]) + self.grid_xy[0] # x, y