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