speed and comments update

2020-03-26 18:34:20 -07:00 · 2020-03-26 18:34:20 -07:00 · dad59220f1
parent faab52913c
commit dad59220f1
1 changed files with 2 additions and 2 deletions
--- a/utils/utils.py
+++ b/utils/utils.py
@ -457,10 +457,10 @@ def build_targets(model, targets):
        t, a = targets, []
        gwh = t[:, 4:6] * ng
        if nt:
-            iou = wh_iou(anchor_vec, gwh)
+            iou = wh_iou(anchor_vec, gwh)  # iou(3,n) = wh_iou(anchor_vec(3,2), gwh(n,2))

            if use_all_anchors:
-                na = len(anchor_vec)  # number of anchors
+                na = anchor_vec.shape[0]  # number of anchors
                a = torch.arange(na).view((-1, 1)).repeat([1, nt]).view(-1)
                t = targets.repeat([na, 1])
                gwh = gwh.repeat([na, 1])