From 9c7dc10b7fecee46e1c54e517c7722faedb93a15 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 10 Apr 2019 16:51:58 +0200 Subject: [PATCH] updates --- utils/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/utils.py b/utils/utils.py index f98917a4..dbe53a1d 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -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)