From cf16f28b299f74742a1409ff5a433846d6ba04f9 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 27 Feb 2021 15:28:23 -0800 Subject: [PATCH] Update test.py (#2319) --- test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test.py b/test.py index 91176eca01db..fd4d339ffea6 100644 --- a/test.py +++ b/test.py @@ -326,6 +326,7 @@ def test(data, test(opt.data, w, opt.batch_size, opt.img_size, 0.25, 0.45, save_json=False, plots=False) elif opt.task == 'study': # run over a range of settings and save/plot + # python test.py --task study --data coco.yaml --iou 0.7 --weights yolov5s.pt yolov5m.pt yolov5l.pt yolov5x.pt x = list(range(256, 1536 + 128, 128)) # x axis (image sizes) for w in opt.weights: f = f'study_{Path(opt.data).stem}_{Path(w).stem}.txt' # filename to save to