Commit 2a43cc7d authored by myownskyW7's avatar myownskyW7
Browse files

high level api minor bugs fix again

parent b4eb18dc
......@@ -34,7 +34,7 @@ def inference_detector(model, imgs, cfg, device='cuda:0'):
img = mmcv.imread(img)
data = _prepare_data(img, img_transform, cfg, device)
with torch.no_grad():
result = model(**data, return_loss=False, rescale=True)
result = model(return_loss=False, rescale=True, **data)
yield result
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment