glenn-jocher commited on
Commit
cd30d83
1 Parent(s): dfeec19

Update test.py (#2319)

Browse files
Files changed (1) hide show
  1. test.py +1 -0
test.py CHANGED
@@ -326,6 +326,7 @@ if __name__ == '__main__':
326
  test(opt.data, w, opt.batch_size, opt.img_size, 0.25, 0.45, save_json=False, plots=False)
327
 
328
  elif opt.task == 'study': # run over a range of settings and save/plot
 
329
  x = list(range(256, 1536 + 128, 128)) # x axis (image sizes)
330
  for w in opt.weights:
331
  f = f'study_{Path(opt.data).stem}_{Path(w).stem}.txt' # filename to save to
 
326
  test(opt.data, w, opt.batch_size, opt.img_size, 0.25, 0.45, save_json=False, plots=False)
327
 
328
  elif opt.task == 'study': # run over a range of settings and save/plot
329
+ # python test.py --task study --data coco.yaml --iou 0.7 --weights yolov5s.pt yolov5m.pt yolov5l.pt yolov5x.pt
330
  x = list(range(256, 1536 + 128, 128)) # x axis (image sizes)
331
  for w in opt.weights:
332
  f = f'study_{Path(opt.data).stem}_{Path(w).stem}.txt' # filename to save to