--verbose bug fix #84

This commit is contained in:
Glenn Jocher committed 2020-06-16 10:22:22 -07:00
1 parent ec81c7b5f2
commit 7c6fd4d670
1 file changed
+4 -3
+4 -3
View File
@@ -17,10 +17,10 @@ def test(data,
save_json=False, save_json=False,
single_cls=False, single_cls=False,
augment=False, augment=False,
verbose=False,
model=None, model=None,
dataloader=None, dataloader=None,
fast=False, fast=False):
verbose=False):
# Initialize/load model and set device # Initialize/load model and set device
if model is None: if model is None:
training = False training = False
@@ -270,7 +270,8 @@ if __name__ == '__main__':
opt.iou_thres, opt.iou_thres,
opt.save_json, opt.save_json,
opt.single_cls, opt.single_cls,
opt.augment) opt.augment,
opt.verbose)
elif opt.task == 'study': # run over a range of settings and save/plot elif opt.task == 'study': # run over a range of settings and save/plot
for weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']: for weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']: