From dfb5d27194faa9efb8ee502504cd293341d79b5b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 16 Jul 2020 12:31:37 -0700 Subject: [PATCH] Update eval.py --- eval.py | 30 +++++++++++++++--------------- 1 file changed, 15 insertions(+), 15 deletions(-) diff --git a/eval.py b/eval.py index eee3cfd..d181712 100644 --- a/eval.py +++ b/eval.py @@ -5,21 +5,21 @@ from models.experimental import * from utils.datasets import * -def evaluate(data, - weights=None, - batch_size=16, - imgsz=640, - conf_thres=0.001, - iou_thres=0.6, # for NMS - save_json=False, - single_cls=False, - augment=False, - verbose=False, - model=None, - dataloader=None, - save_dir='', - merge=False, - save_txt=False): +def test(data, + weights=None, + batch_size=16, + imgsz=640, + conf_thres=0.001, + iou_thres=0.6, # for NMS + save_json=False, + single_cls=False, + augment=False, + verbose=False, + model=None, + dataloader=None, + save_dir='', + merge=False, + save_txt=False): # Initialize/load model and set device training = model is not None if training: # called by train.py