diff --git a/mmdet/apis/inference.py b/mmdet/apis/inference.py index 67b2b241404251dc966a9e9322b3ee6975c35d4c..ccf228a77eb42266dd5e61cc7860ca4af34d7036 100644 --- 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