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

speed and comments update

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

4
utils/utils.py
View File

@ -457,10 +457,10 @@ def build_targets(model, targets):
t, a = targets, [] t, a = targets, []
gwh = t[:, 4:6] * ng gwh = t[:, 4:6] * ng
if nt: 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: 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) a = torch.arange(na).view((-1, 1)).repeat([1, nt]).view(-1)
t = targets.repeat([na, 1]) t = targets.repeat([na, 1])
gwh = gwh.repeat([na, 1]) gwh = gwh.repeat([na, 1])