fix faster rcnn use im_shape (#2779)

c3d8b727 · Kaipeng Deng · GitHub · 80b47b1b · c3d8b727
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

ppdet/modeling/architectures/faster_rcnn.py ppdet/modeling/architectures/faster_rcnn.py +1 -1

未找到文件。
--- 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())