--rect test bug patch

2020-07-09 20:36:30 -07:00 · 2020-07-09 20:36:30 -07:00 · d40e4781f1
parent 520f5de6f0
commit d40e4781f1
1 changed files with 1 additions and 1 deletions
--- a/test.py
+++ b/test.py
@ -59,7 +59,7 @@ def test(data,
        _ = model(img.half() if half else img) if device.type != 'cpu' else None  # run once
        path = data['test'] if opt.task == 'test' else data['val']  # path to val/test images
        dataloader = create_dataloader(path, imgsz, batch_size, model.stride.max(), opt,
-                                       hyp=None, augment=False, cache=False, pad=0.5, rect=True)[0]
+                                       hyp=None, augment=False, cache=False, pad=0.5, rect=False)[0]

    seen = 0
    names = model.names if hasattr(model, 'names') else model.module.names