瀏覽代碼

--verbose bug fix #84

5.0
Glenn Jocher 4 年之前
父節點
當前提交
7c6fd4d670
共有 1 個檔案被更改,包括 4 行新增3 行删除
  1. +4
    -3
      test.py

+ 4
- 3
test.py 查看文件

@@ -17,10 +17,10 @@ def test(data,
save_json=False,
single_cls=False,
augment=False,
verbose=False,
model=None,
dataloader=None,
fast=False,
verbose=False):
fast=False):
# Initialize/load model and set device
if model is None:
training = False
@@ -270,7 +270,8 @@ if __name__ == '__main__':
opt.iou_thres,
opt.save_json,
opt.single_cls,
opt.augment)
opt.augment,
opt.verbose)

elif opt.task == 'study': # run over a range of settings and save/plot
for weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']:

Loading…
取消
儲存