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
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=
|
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
|