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

Update eval.py

上级 b4faccab
...@@ -5,21 +5,21 @@ from models.experimental import * ...@@ -5,21 +5,21 @@ from models.experimental import *
from utils.datasets import * from utils.datasets import *
def evaluate(data, def test(data,
weights=None, weights=None,
batch_size=16, batch_size=16,
imgsz=640, imgsz=640,
conf_thres=0.001, conf_thres=0.001,
iou_thres=0.6, # for NMS iou_thres=0.6, # for NMS
save_json=False, save_json=False,
single_cls=False, single_cls=False,
augment=False, augment=False,
verbose=False, verbose=False,
model=None, model=None,
dataloader=None, dataloader=None,
save_dir='', save_dir='',
merge=False, merge=False,
save_txt=False): save_txt=False):
# Initialize/load model and set device # Initialize/load model and set device
training = model is not None training = model is not None
if training: # called by train.py if training: # called by train.py
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论