[Fix] Fix KeyError: 'ori_filename' when using --show-dir with centern…

…et_resnet18_dcnv2_140e_coco.py (open-mmlab#7865)
ZwwWayne · Jul 19, 2022 · 653ec76 · 653ec76
1 parent 68c8994
commit 653ec76
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/configs/centernet/centernet_resnet18_dcnv2_140e_coco.py b/configs/centernet/centernet_resnet18_dcnv2_140e_coco.py
@@ -78,9 +78,9 @@
             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'])
         ])
 ]