diff --git a/test.py b/test.py index 8d252ff..1f937a7 100644 --- a/test.py +++ b/test.py @@ -17,10 +17,10 @@ def test(data, save_json=False, single_cls=False, augment=False, + verbose=False, model=None, dataloader=None, - fast=False, - verbose=False): + fast=False): # Initialize/load model and set device if model is None: training = False @@ -270,7 +270,8 @@ if __name__ == '__main__': opt.iou_thres, opt.save_json, opt.single_cls, - opt.augment) + opt.augment, + opt.verbose) elif opt.task == 'study': # run over a range of settings and save/plot for weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']: