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

* Update val.py `pad = 0.0 if task == speed else 0.5` * Cleanup
fhkiel-mlaip · Oct 11, 2021 · 488ad2a · 488ad2a
1 parent bc0360b
commit 488ad2a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/val.py b/val.py
@@ -147,8 +147,9 @@ def run(data,
     if not training:
         if device.type != 'cpu':
             model(torch.zeros(1, 3, imgsz, imgsz).to(device).type_as(next(model.parameters())))  # run once
+        pad = 0.0 if task == 'speed' else 0.5
         task = task if task in ('train', 'val', 'test') else 'val'  # path to train/val/test images
-        dataloader = create_dataloader(data[task], imgsz, batch_size, gs, single_cls, pad=0.5, rect=True,
+        dataloader = create_dataloader(data[task], imgsz, batch_size, gs, single_cls, pad=pad, rect=True,
                                        prefix=colorstr(f'{task}: '))[0]
 
     seen = 0