updates
This commit is contained in:
parent
9b32d3cb5d
commit
be10b75eb4
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue