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:
......
......@@ -117,4 +117,4 @@ def train_detector(model, dataset, cfg):
runner.resume(cfg.resume_from)
elif cfg.load_from:
runner.load_checkpoint(cfg.load_from)
runner.run(data_loaders, cfg.workflow, cfg.total_epochs)
\ No newline at end of file
runner.run(data_loaders, cfg.workflow, cfg.total_epochs)
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