Skip to content

GitLab

Commit b4eb18dc authored by myownskyW7's avatar myownskyW7
Browse files

high level api minor bugs fix

parent 4c1da636
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmdet/api/inference.py
View file @ b4eb18dc
...@@ -27,7 +27,7 @@ def inference_detector(model, imgs, cfg, device='cuda:0'): ...@@ -27,7 +27,7 @@ def inference_detector(model, imgs, cfg, device='cuda:0'):
imgs = imgs if isinstance(imgs, list) else [imgs] imgs = imgs if isinstance(imgs, list) else [imgs]
img_transform = ImageTransform( 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 = model.to(device)
model.eval() model.eval()
for img in imgs: 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