From 2a43cc7d9add9b48f2c88788850adeea6f67b83e Mon Sep 17 00:00:00 2001
From: myownskyW7 <727032989@qq.com>
Date: Thu, 11 Oct 2018 23:40:40 +0800
Subject: [PATCH] high level api minor bugs fix again
---
mmdet/api/inference.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mmdet/api/inference.py b/mmdet/api/inference.py
index d452c66..0addd59 100644
--- a/mmdet/api/inference.py
+++ b/mmdet/api/inference.py
@@ -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
--
GitLab