From e0833ed21e23ee21b9b4f67c1ce6e68f70398c01 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 22 Dec 2019 13:07:00 -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 e3f1f7e5..02bb8b2b 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -444,7 +444,7 @@ def build_targets(model, targets): gi, gj = gxy.long().t() # grid x, y indices indices.append((b, a, gj, gi)) - # GIoU + # Box gxy -= gxy.floor() # xy tbox.append(torch.cat((gxy, gwh), 1)) # xywh (grids) av.append(anchor_vec[a]) # anchor vec