glenn-jocher commited on
Commit
7c6fd4d
1 Parent(s): ec81c7b

--verbose bug fix #84

Browse files
Files changed (1) hide show
  1. test.py +4 -3
test.py CHANGED
@@ -17,10 +17,10 @@ def test(data,
17
  save_json=False,
18
  single_cls=False,
19
  augment=False,
 
20
  model=None,
21
  dataloader=None,
22
- fast=False,
23
- verbose=False):
24
  # Initialize/load model and set device
25
  if model is None:
26
  training = False
@@ -270,7 +270,8 @@ if __name__ == '__main__':
270
  opt.iou_thres,
271
  opt.save_json,
272
  opt.single_cls,
273
- opt.augment)
 
274
 
275
  elif opt.task == 'study': # run over a range of settings and save/plot
276
  for weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']:
 
17
  save_json=False,
18
  single_cls=False,
19
  augment=False,
20
+ verbose=False,
21
  model=None,
22
  dataloader=None,
23
+ fast=False):
 
24
  # Initialize/load model and set device
25
  if model is None:
26
  training = False
 
270
  opt.iou_thres,
271
  opt.save_json,
272
  opt.single_cls,
273
+ opt.augment,
274
+ opt.verbose)
275
 
276
  elif opt.task == 'study': # run over a range of settings and save/plot
277
  for weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']: