glenn-jocher
commited on
Commit
•
d40e478
1
Parent(s):
520f5de
--rect test bug patch
Browse files
test.py
CHANGED
@@ -59,7 +59,7 @@ def test(data,
|
|
59 |
_ = model(img.half() if half else img) if device.type != 'cpu' else None # run once
|
60 |
path = data['test'] if opt.task == 'test' else data['val'] # path to val/test images
|
61 |
dataloader = create_dataloader(path, imgsz, batch_size, model.stride.max(), opt,
|
62 |
-
hyp=None, augment=False, cache=False, pad=0.5, rect=
|
63 |
|
64 |
seen = 0
|
65 |
names = model.names if hasattr(model, 'names') else model.module.names
|
|
|
59 |
_ = model(img.half() if half else img) if device.type != 'cpu' else None # run once
|
60 |
path = data['test'] if opt.task == 'test' else data['val'] # path to val/test images
|
61 |
dataloader = create_dataloader(path, imgsz, batch_size, model.stride.max(), opt,
|
62 |
+
hyp=None, augment=False, cache=False, pad=0.5, rect=False)[0]
|
63 |
|
64 |
seen = 0
|
65 |
names = model.names if hasattr(model, 'names') else model.module.names
|