Unverified 提交 cd30d838 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Update test.py (#2319)

上级 dfeec198
...@@ -326,6 +326,7 @@ if __name__ == '__main__': ...@@ -326,6 +326,7 @@ if __name__ == '__main__':
test(opt.data, w, opt.batch_size, opt.img_size, 0.25, 0.45, save_json=False, plots=False) 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 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) x = list(range(256, 1536 + 128, 128)) # x axis (image sizes)
for w in opt.weights: for w in opt.weights:
f = f'study_{Path(opt.data).stem}_{Path(w).stem}.txt' # filename to save to f = f'study_{Path(opt.data).stem}_{Path(w).stem}.txt' # filename to save to
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论