Explorar el Código

--rect test bug patch

5.0
Glenn Jocher hace 4 años
padre
commit
d40e4781f1
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  1. +1
    -1
      test.py

+ 1
- 1
test.py Ver fichero

@@ -59,7 +59,7 @@ def test(data,
_ = model(img.half() if half else img) if device.type != 'cpu' else None # run once
path = data['test'] if opt.task == 'test' else data['val'] # path to val/test images
dataloader = create_dataloader(path, imgsz, batch_size, model.stride.max(), opt,
hyp=None, augment=False, cache=False, pad=0.5, rect=True)[0]
hyp=None, augment=False, cache=False, pad=0.5, rect=False)[0]

seen = 0
names = model.names if hasattr(model, 'names') else model.module.names

Cargando…
Cancelar
Guardar