Update eval.py

This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-07-16 12:31:37 -07:00
1 parent b4faccabac
commit dfb5d27194
1 file changed
+15 -15
+15 -15
View File
@@ -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