Update eval.py
This commit is contained in:
1 parent
b4faccabac
commit
dfb5d27194
1 file changed
+15
-15
@@ -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
|
||||||
|
|||||||
Reference in new issue
Block a user