Update eval.py

pull/1/head
Glenn Jocher 5 years ago committed by GitHub
parent b4faccabac
commit dfb5d27194
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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

Loading…
Cancel
Save