diff --git a/ppdet/modeling/architectures/faster_rcnn.py b/ppdet/modeling/architectures/faster_rcnn.py index 58d1e7978e3fc17356e7ce67c6b0d999909f435d..641ae0708bdda9be637861bfd29e43a6bc3b5bab 100644 --- a/ppdet/modeling/architectures/faster_rcnn.py +++ b/ppdet/modeling/architectures/faster_rcnn.py @@ -64,7 +64,7 @@ class FasterRCNN(object): gt_box = feed_vars['gt_box'] is_crowd = feed_vars['is_crowd'] else: - im_shape = feed_vars['im_info'] + im_shape = feed_vars['im_shape'] body_feats = self.backbone(im) body_feat_names = list(body_feats.keys())