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

updates

This commit is contained in:
Glenn Jocher 2019-03-04 16:01:23 +01:00
parent 175e231c55
commit 5ca987eaed
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

@ -243,7 +243,7 @@ def build_targets(target, anchor_vec, nA, nC, nG):
tcls = torch.ByteTensor(nB, nA, nG, nG, nC).fill_(0) # nC = number of classes tcls = torch.ByteTensor(nB, nA, nG, nG, nC).fill_(0) # nC = number of classes
for b in range(nB): for b in range(nB):
t = target[b] t = target[b].cpu()
nTb = len(t) # number of targets nTb = len(t) # number of targets
if nTb == 0: if nTb == 0:
continue continue
@ -257,7 +257,7 @@ def build_targets(target, anchor_vec, nA, nC, nG):
box1 = gwh box1 = gwh
box2 = anchor_vec.unsqueeze(1) box2 = anchor_vec.unsqueeze(1)
print(box1.device,box2.device) print(box1.device, box2.device)
inter_area = torch.min(box1, box2).prod(2) inter_area = torch.min(box1, box2).prod(2)
iou = inter_area / (box1.prod(1) + box2.prod(2) - inter_area + 1e-16) iou = inter_area / (box1.prod(1) + box2.prod(2) - inter_area + 1e-16)