Unverified 提交 7d37b3c5 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

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

* Update val.py `pad = 0.0 if task == speed else 0.5` * Cleanup
上级 ba4b79de
...@@ -147,8 +147,9 @@ def run(data, ...@@ -147,8 +147,9 @@ def run(data,
if not training: if not training:
if device.type != 'cpu': if device.type != 'cpu':
model(torch.zeros(1, 3, imgsz, imgsz).to(device).type_as(next(model.parameters()))) # run once 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 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] prefix=colorstr(f'{task}: '))[0]
seen = 0 seen = 0
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论