glenn-jocher commited on
Commit
7d37b3c
1 Parent(s): ba4b79d

Update val.py `pad = 0.0 if task == speed else 0.5` (#5121)

Browse files

* Update val.py `pad = 0.0 if task == speed else 0.5`

* Cleanup

Files changed (1) hide show
  1. val.py +2 -1
val.py CHANGED
@@ -147,8 +147,9 @@ def run(data,
147
  if not training:
148
  if device.type != 'cpu':
149
  model(torch.zeros(1, 3, imgsz, imgsz).to(device).type_as(next(model.parameters()))) # run once
 
150
  task = task if task in ('train', 'val', 'test') else 'val' # path to train/val/test images
151
- dataloader = create_dataloader(data[task], imgsz, batch_size, gs, single_cls, pad=0.5, rect=True,
152
  prefix=colorstr(f'{task}: '))[0]
153
 
154
  seen = 0
 
147
  if not training:
148
  if device.type != 'cpu':
149
  model(torch.zeros(1, 3, imgsz, imgsz).to(device).type_as(next(model.parameters()))) # run once
150
+ pad = 0.0 if task == 'speed' else 0.5
151
  task = task if task in ('train', 'val', 'test') else 'val' # path to train/val/test images
152
+ dataloader = create_dataloader(data[task], imgsz, batch_size, gs, single_cls, pad=pad, rect=True,
153
  prefix=colorstr(f'{task}: '))[0]
154
 
155
  seen = 0