Commit b4eb18dc authored by myownskyW7's avatar myownskyW7
Browse files

high level api minor bugs fix

parent 4c1da636
......@@ -27,7 +27,7 @@ def inference_detector(model, imgs, cfg, device='cuda:0'):
imgs = imgs if isinstance(imgs, list) else [imgs]
img_transform = ImageTransform(
**cfg.img_norm_cfg, size_divisor=cfg.data.test.size_divisor)
size_divisor=cfg.data.test.size_divisor, **cfg.img_norm_cfg)
model = model.to(device)
model.eval()
for img in imgs:
......
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