Update eval.py
Browse files
eval.py
CHANGED
@@ -5,21 +5,21 @@ from models.experimental import *
|
|
5 |
from utils.datasets import *
|
6 |
|
7 |
|
8 |
-
def
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
# Initialize/load model and set device
|
24 |
training = model is not None
|
25 |
if training: # called by train.py
|
|
|
5 |
from utils.datasets import *
|
6 |
|
7 |
|
8 |
+
def test(data,
|
9 |
+
weights=None,
|
10 |
+
batch_size=16,
|
11 |
+
imgsz=640,
|
12 |
+
conf_thres=0.001,
|
13 |
+
iou_thres=0.6, # for NMS
|
14 |
+
save_json=False,
|
15 |
+
single_cls=False,
|
16 |
+
augment=False,
|
17 |
+
verbose=False,
|
18 |
+
model=None,
|
19 |
+
dataloader=None,
|
20 |
+
save_dir='',
|
21 |
+
merge=False,
|
22 |
+
save_txt=False):
|
23 |
# Initialize/load model and set device
|
24 |
training = model is not None
|
25 |
if training: # called by train.py
|