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

updates

This commit is contained in:
Glenn Jocher 2019-08-18 21:28:49 +02:00
parent b779e6ef69
commit ebd93e354d
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

@ -327,7 +327,7 @@ def compute_loss(p, targets, model): # predictions, targets, model
bs = p[0].shape[0] # batch size bs = p[0].shape[0] # batch size
k = bs / 64 # loss gain k = bs / 64 # loss gain
arc = 'normal' # (normal, uCE, uBCE, uBCEs) detection architectures arc = 'normal' # (normal, uCE, uBCE, uBCEs) detection architectures
for i, pi0 in enumerate(p): # layer i predictions, i for i, pi in enumerate(p): # layer index, layer predictions
b, a, gj, gi = indices[i] # image, anchor, gridy, gridx b, a, gj, gi = indices[i] # image, anchor, gridy, gridx
tobj = torch.zeros_like(pi[..., 0]) # target obj tobj = torch.zeros_like(pi[..., 0]) # target obj