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