提交 c3d8b727 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix faster rcnn use im_shape (#2779)

上级 80b47b1b
...@@ -64,7 +64,7 @@ class FasterRCNN(object): ...@@ -64,7 +64,7 @@ class FasterRCNN(object):
gt_box = feed_vars['gt_box'] gt_box = feed_vars['gt_box']
is_crowd = feed_vars['is_crowd'] is_crowd = feed_vars['is_crowd']
else: else:
im_shape = feed_vars['im_info'] im_shape = feed_vars['im_shape']
body_feats = self.backbone(im) body_feats = self.backbone(im)
body_feat_names = list(body_feats.keys()) body_feat_names = list(body_feats.keys())
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册