fix test api (#1255)

c3acdfd1 · Kai Chen · GitHub · c0a4c007 · c3acdfd1
Unverified Commit c3acdfd1 authored Aug 25, 2019 by Kai Chen Committed by GitHub Aug 25, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

mmdet/apis/inference.py mmdet/apis/inference.py +5 -0

No files found.
--- a/mmdet/apis/inference.py
+++ b/mmdet/apis/inference.py
@@ -49,8 +49,13 @@ def init_detector(config, checkpoint=None, device='cuda:0'):
 class LoadImage(object):

    def __call__(self, results):
+        if isinstance(results['img'], str):
+            results['filename'] = results['img']
+        else:
+            results['filename'] = None
        img = mmcv.imread(results['img'])
        results['img'] = img
+        results['img_shape'] = img.shape
        results['ori_shape'] = img.shape
        return results