Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge NMS update

This commit is contained in:
Glenn Jocher 2020-03-26 18:41:04 -07:00
parent dad59220f1
commit 01ee0c5e95
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/utils.py
View File

@ -442,7 +442,7 @@ def compute_loss(p, targets, model): # predictions, targets, model
def build_targets(model, targets): def build_targets(model, targets):
# targets = [image, class, x, y, w, h] # targets = [image, class, x, y, w, h]
nt = len(targets) nt = targets.shape[0]
tcls, tbox, indices, av = [], [], [], [] tcls, tbox, indices, av = [], [], [], []
multi_gpu = type(model) in (nn.parallel.DataParallel, nn.parallel.DistributedDataParallel) multi_gpu = type(model) in (nn.parallel.DataParallel, nn.parallel.DistributedDataParallel)
reject, use_all_anchors = True, True reject, use_all_anchors = True, True