From 653ec76935da1d3d75895b83cd48f0c918363fac Mon Sep 17 00:00:00 2001 From: jbwang1997 Date: Thu, 28 Apr 2022 21:18:49 +0800 Subject: [PATCH] [Fix] Fix KeyError: 'ori_filename' when using --show-dir with centernet_resnet18_dcnv2_140e_coco.py (#7865) --- configs/centernet/centernet_resnet18_dcnv2_140e_coco.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/configs/centernet/centernet_resnet18_dcnv2_140e_coco.py b/configs/centernet/centernet_resnet18_dcnv2_140e_coco.py index cb225ef7af3..b8a0bb10ddc 100644 --- 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']) ]) ]