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

augmented testing

This commit is contained in:
Glenn Jocher 2020-03-26 11:25:44 -07:00
parent 23b34f4db8
commit c71ab7d506
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
test.py
View File

@ -87,8 +87,7 @@ def test(cfg,
# Disable gradients # Disable gradients
with torch.no_grad(): with torch.no_grad():
aug = False # augment https://github.com/ultralytics/yolov3/issues/931 if opt.augment: # augmented testing https://github.com/ultralytics/yolov3/issues/931
if aug:
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,7 +98,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
if aug: if opt.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
x[2][..., :4] /= 0.7 # scale x[2][..., :4] /= 0.7 # scale
@ -247,6 +246,7 @@ if __name__ == '__main__':
parser.add_argument('--task', default='test', help="'test', 'study', 'benchmark'") parser.add_argument('--task', default='test', help="'test', 'study', 'benchmark'")
parser.add_argument('--device', default='', help='device id (i.e. 0 or 0,1) or cpu') parser.add_argument('--device', default='', help='device id (i.e. 0 or 0,1) or cpu')
parser.add_argument('--single-cls', action='store_true', help='train as single-class dataset') parser.add_argument('--single-cls', action='store_true', help='train as single-class dataset')
parser.add_argument('--augment', action='store_true', help='augmented testing')
opt = parser.parse_args() opt = parser.parse_args()
opt.save_json = opt.save_json or any([x in opt.data for x in ['coco.data', 'coco2014.data', 'coco2017.data']]) opt.save_json = opt.save_json or any([x in opt.data for x in ['coco.data', 'coco2014.data', 'coco2017.data']])
print(opt) print(opt)