diff --git a/demo/mono_det_demo.py b/demo/mono_det_demo.py index 42416fd9f4..6e099927db 100644 --- a/demo/mono_det_demo.py +++ b/demo/mono_det_demo.py @@ -57,7 +57,6 @@ def parse_args(): call_args['inputs'] = dict( img=call_args.pop('img'), infos=call_args.pop('infos')) - call_args.pop('cam_type') if call_args['no_save_vis'] and call_args['no_save_pred']: call_args['out_dir'] = '' diff --git a/demo/multi_modality_demo.py b/demo/multi_modality_demo.py index e3c28cdc48..4b3dbd4f86 100644 --- a/demo/multi_modality_demo.py +++ b/demo/multi_modality_demo.py @@ -60,7 +60,6 @@ def parse_args(): points=call_args.pop('pcd'), img=call_args.pop('img'), infos=call_args.pop('infos')) - call_args.pop('cam_type') if call_args['no_save_vis'] and call_args['no_save_pred']: call_args['out_dir'] = '' diff --git a/mmdet3d/apis/inferencers/base_3d_inferencer.py b/mmdet3d/apis/inferencers/base_3d_inferencer.py index 6564325e8c..8b72da1a3e 100644 --- a/mmdet3d/apis/inferencers/base_3d_inferencer.py +++ b/mmdet3d/apis/inferencers/base_3d_inferencer.py @@ -167,6 +167,7 @@ def _dispatch_kwargs(self, kwargs['img_out_dir'] = out_dir kwargs['pred_out_dir'] = out_dir if cam_type != '': + kwargs['cam_type'] = cam_type kwargs['cam_type_dir'] = cam_type return super()._dispatch_kwargs(**kwargs)