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

updates

This commit is contained in:
Glenn Jocher 2019-12-22 13:07:00 -08:00
parent a96285870d
commit e0833ed21e
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

@ -444,7 +444,7 @@ def build_targets(model, targets):
gi, gj = gxy.long().t() # grid x, y indices gi, gj = gxy.long().t() # grid x, y indices
indices.append((b, a, gj, gi)) indices.append((b, a, gj, gi))
# GIoU # Box
gxy -= gxy.floor() # xy gxy -= gxy.floor() # xy
tbox.append(torch.cat((gxy, gwh), 1)) # xywh (grids) tbox.append(torch.cat((gxy, gwh), 1)) # xywh (grids)
av.append(anchor_vec[a]) # anchor vec av.append(anchor_vec[a]) # anchor vec