updates

2019-12-22 13:07:00 -08:00 · 2019-12-22 13:07:00 -08:00 · e0833ed21e
parent a96285870d
commit e0833ed21e
1 changed files with 1 additions and 1 deletions
--- 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