glenn-jocher commited on
Commit
d40e478
1 Parent(s): 520f5de

--rect test bug patch

Browse files
Files changed (1) hide show
  1. test.py +1 -1
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=True)[0]
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