glenn-jocher
commited on
Commit
•
cd30d83
1
Parent(s):
dfeec19
Update test.py (#2319)
Browse files
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
|