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

updates

This commit is contained in:
Glenn Jocher 2019-04-10 16:51:58 +02:00
parent f0b4f9f4fb
commit 9c7dc10b7f
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

@ -314,8 +314,8 @@ def build_targets(model, targets):
txy.append(gxy - gxy.floor())
# Width and height
twh.append(torch.log(gwh / layer.anchor_vec[a])) # yolo method
# twh.append((gwh / layer.anchor_vec[a]) ** (1 / 3) / 2) # power method
twh.append(torch.log(gwh / layer.anchor_vec[a])) # wh yolo method
# twh.append((gwh / layer.anchor_vec[a]) ** (1 / 3) / 2) # wh power method
# Class
tcls.append(c)