glenn-jocher commited on
Commit
dfb5d27
·
unverified ·
1 Parent(s): b4facca

Update eval.py

Browse files
Files changed (1) hide show
  1. eval.py +15 -15
eval.py CHANGED
@@ -5,21 +5,21 @@ from models.experimental import *
5
  from utils.datasets import *
6
 
7
 
8
- def evaluate(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
 
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