Update test.py (ultralytics#2319)

taicaile · Feb 27, 2021 · de26526 · de26526
1 parent a2367f3
commit de26526
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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