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

test augmentation comments

This commit is contained in:
Glenn Jocher 2020-03-26 11:34:32 -07:00
parent f91b1fb13a
commit 94344f5bea
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test.py
View File

@ -88,7 +88,8 @@ def test(cfg,
# Disable gradients # Disable gradients
with torch.no_grad(): with torch.no_grad():
if augment: # augmented testing https://github.com/ultralytics/yolov3/issues/931 # Augment images
if augment: # https://github.com/ultralytics/yolov3/issues/931
imgs = torch.cat((imgs, imgs = torch.cat((imgs,
imgs.flip(3), # flip-lr imgs.flip(3), # flip-lr
torch_utils.scale_img(imgs, 0.7), # scale torch_utils.scale_img(imgs, 0.7), # scale
@ -99,6 +100,7 @@ def test(cfg,
inf_out, train_out = model(imgs) # inference and training outputs inf_out, train_out = model(imgs) # inference and training outputs
t0 += torch_utils.time_synchronized() - t t0 += torch_utils.time_synchronized() - t
# De-augment results
if augment: if augment:
x = torch.split(inf_out, nb, dim=0) x = torch.split(inf_out, nb, dim=0)
x[1][..., 0] = width - x[1][..., 0] # flip lr x[1][..., 0] = width - x[1][..., 0] # flip lr