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 15:59:11 +01:00
parent 0b3a17362c
commit 175e231c55
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -150,8 +150,6 @@ class YOLOLayer(nn.Module):
p_conf = p[..., 4] # Conf p_conf = p[..., 4] # Conf
p_cls = p[..., 5:] # Class p_cls = p[..., 5:] # Class
print(self.anchor_vec.device, self.anchor_vec, self.nA, self.nC, nG)
txy, twh, mask, tcls = build_targets(targets, self.anchor_vec, self.nA, self.nC, nG) txy, twh, mask, tcls = build_targets(targets, self.anchor_vec, self.nA, self.nC, nG)
tcls = tcls[mask] tcls = tcls[mask]

3
utils/utils.py
View File

@ -256,6 +256,9 @@ def build_targets(target, anchor_vec, nA, nC, nG):
# iou of targets-anchors (using wh only) # iou of targets-anchors (using wh only)
box1 = gwh box1 = gwh
box2 = anchor_vec.unsqueeze(1) box2 = anchor_vec.unsqueeze(1)
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)