[Fix] Fix KeyError: 'ori_filename' when using --show-dir with centernet_resnet18_dcnv2_140e_coco.py (#7865)

3 years ago · e283bae020
parent 41ba2f4535
commit e283bae020
1 changed files with 3 additions and 3 deletions
--- a/configs/centernet/centernet_resnet18_dcnv2_140e_coco.py
+++ b/configs/centernet/centernet_resnet18_dcnv2_140e_coco.py
@ -78,9 +78,9 @@ test_pipeline = [
            dict(type='DefaultFormatBundle'),
            dict(
                type='Collect',
-                meta_keys=('filename', 'ori_shape', 'img_shape', 'pad_shape',
-                           'scale_factor', 'flip', 'flip_direction',
-                           'img_norm_cfg', 'border'),
+                meta_keys=('filename', 'ori_filename', 'ori_shape',
+                           'img_shape', 'pad_shape', 'scale_factor', 'flip',
+                           'flip_direction', 'img_norm_cfg', 'border'),
                keys=['img'])
        ])
 ]