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

cleanup

This commit is contained in:
Glenn Jocher 2020-04-14 04:15:53 -07:00
parent 198a5a591d
commit 1681249588
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

@ -443,7 +443,7 @@ def build_targets(p, targets, model):
anchor_vec = model.module.module_list[j].anchor_vec if multi_gpu else model.module_list[j].anchor_vec anchor_vec = model.module.module_list[j].anchor_vec if multi_gpu else model.module_list[j].anchor_vec
# iou of targets-anchors # iou of targets-anchors
gain[2:] = torch.tensor(p[j].shape)[[2, 3, 2, 3]] # xyxy gain gain[2:] = torch.tensor(p[i].shape)[[2, 3, 2, 3]] # xyxy gain
t, a = targets * gain, [] t, a = targets * gain, []
gwh = t[:, 4:6] gwh = t[:, 4:6]
if nt: if nt: