From c3d8b72724f571f0d39c62698e9d92388bdad027 Mon Sep 17 00:00:00 2001 From: Kaipeng Deng Date: Wed, 10 Jul 2019 16:01:34 +0800 Subject: [PATCH] fix faster rcnn use im_shape (#2779) --- ppdet/modeling/architectures/faster_rcnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/modeling/architectures/faster_rcnn.py b/ppdet/modeling/architectures/faster_rcnn.py index 58d1e7978..641ae0708 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()) -- GitLab