This commit is contained in:
Glenn Jocher 2019-04-02 20:21:00 +02:00
parent 9b32d3cb5d
commit be10b75eb4
1 changed files with 1 additions and 1 deletions

View File

@ -321,7 +321,7 @@ def build_targets(model, targets):
# Width and height # Width and height
twh.append(torch.log(gwh / layer.anchor_vec[a])) # yolo method twh.append(torch.log(gwh / layer.anchor_vec[a])) # yolo method
# twh.append(torch.sqrt(gwh / anchor_vec[a]) / 2) # power method # twh.append(torch.sqrt(gwh / layer.anchor_vec[a]) / 2) # power method
# Class # Class
tcls.append(c) tcls.append(c)