From 01ee0c5e95593e5f2c9143b2872096f43ab28069 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 26 Mar 2020 18:41:04 -0700 Subject: [PATCH] Merge NMS update --- utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/utils.py b/utils/utils.py index d3a7ecf5..f023893c 100755 --- 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