glenn-jocher commited on
Commit
4ef9261
·
unverified ·
1 Parent(s): 5948f20

Remove redundant speed/study `half` argument (#3557)

Browse files
Files changed (1) hide show
  1. test.py +2 -3
test.py CHANGED
@@ -336,8 +336,7 @@ if __name__ == '__main__':
336
 
337
  elif opt.task == 'speed': # speed benchmarks
338
  for w in opt.weights if isinstance(opt.weights, list) else [opt.weights]:
339
- test(opt.data, w, opt.batch_size, opt.img_size, 0.25, 0.45, save_json=False, plots=False, half=True,
340
- opt=opt)
341
 
342
  elif opt.task == 'study': # run over a range of settings and save/plot
343
  # python test.py --task study --data coco.yaml --iou 0.7 --weights yolov5s.pt yolov5m.pt yolov5l.pt yolov5x.pt
@@ -348,7 +347,7 @@ if __name__ == '__main__':
348
  for i in x: # img-size
349
  print(f'\nRunning {f} point {i}...')
350
  r, _, t = test(opt.data, w, opt.batch_size, i, opt.conf_thres, opt.iou_thres, opt.save_json,
351
- plots=False, half=True, opt=opt)
352
  y.append(r + t) # results and times
353
  np.savetxt(f, y, fmt='%10.4g') # save
354
  os.system('zip -r study.zip study_*.txt')
 
336
 
337
  elif opt.task == 'speed': # speed benchmarks
338
  for w in opt.weights if isinstance(opt.weights, list) else [opt.weights]:
339
+ test(opt.data, w, opt.batch_size, opt.img_size, 0.25, 0.45, save_json=False, plots=False, opt=opt)
 
340
 
341
  elif opt.task == 'study': # run over a range of settings and save/plot
342
  # python test.py --task study --data coco.yaml --iou 0.7 --weights yolov5s.pt yolov5m.pt yolov5l.pt yolov5x.pt
 
347
  for i in x: # img-size
348
  print(f'\nRunning {f} point {i}...')
349
  r, _, t = test(opt.data, w, opt.batch_size, i, opt.conf_thres, opt.iou_thres, opt.save_json,
350
+ plots=False, opt=opt)
351
  y.append(r + t) # results and times
352
  np.savetxt(f, y, fmt='%10.4g') # save
353
  os.system('zip -r study.zip study_*.txt')