glenn-jocher
commited on
Commit
•
7c6fd4d
1
Parent(s):
ec81c7b
--verbose bug fix #84
Browse files
test.py
CHANGED
@@ -17,10 +17,10 @@ def test(data,
|
|
17 |
save_json=False,
|
18 |
single_cls=False,
|
19 |
augment=False,
|
|
|
20 |
model=None,
|
21 |
dataloader=None,
|
22 |
-
fast=False
|
23 |
-
verbose=False):
|
24 |
# Initialize/load model and set device
|
25 |
if model is None:
|
26 |
training = False
|
@@ -270,7 +270,8 @@ if __name__ == '__main__':
|
|
270 |
opt.iou_thres,
|
271 |
opt.save_json,
|
272 |
opt.single_cls,
|
273 |
-
opt.augment
|
|
|
274 |
|
275 |
elif opt.task == 'study': # run over a range of settings and save/plot
|
276 |
for weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']:
|
|
|
17 |
save_json=False,
|
18 |
single_cls=False,
|
19 |
augment=False,
|
20 |
+
verbose=False,
|
21 |
model=None,
|
22 |
dataloader=None,
|
23 |
+
fast=False):
|
|
|
24 |
# Initialize/load model and set device
|
25 |
if model is None:
|
26 |
training = False
|
|
|
270 |
opt.iou_thres,
|
271 |
opt.save_json,
|
272 |
opt.single_cls,
|
273 |
+
opt.augment,
|
274 |
+
opt.verbose)
|
275 |
|
276 |
elif opt.task == 'study': # run over a range of settings and save/plot
|
277 |
for weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']:
|