Merge NMS update

2020-03-26 18:41:04 -07:00 · 2020-03-26 18:41:04 -07:00 · 01ee0c5e95
parent dad59220f1
commit 01ee0c5e95
1 changed files with 1 additions and 1 deletions
--- a/utils/utils.py
+++ b/utils/utils.py
@ -442,7 +442,7 @@ def compute_loss(p, targets, model):  # predictions, targets, model
 def build_targets(model, targets):
    # targets = [image, class, x, y, w, h]

-    nt = len(targets)
+    nt = targets.shape[0]
    tcls, tbox, indices, av = [], [], [], []
    multi_gpu = type(model) in (nn.parallel.DataParallel, nn.parallel.DistributedDataParallel)
    reject, use_all_anchors = True, True