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 *
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