未验证 提交 274d953c 编写于 作者: W wangguanzhong 提交者: GitHub

fix rpn_only in cascade mask fpn (#3231)

* fix rpn_only in cascade mask fpn

* fix rpn_only in cascade rcnn
上级 8f83f81c
......@@ -114,8 +114,8 @@ class CascadeMaskRCNN(object):
if self.rpn_only:
im_scale = fluid.layers.slice(
im_info, [1], starts=[2], ends=[3])
im_scale = fluid.layers.sequence_expand(im_scale, rois)
rois = rois / im_scale
im_scale = fluid.layers.sequence_expand(im_scale, rpn_rois)
rois = rpn_rois / im_scale
return {'proposal': rois}
proposal_list = []
......
......@@ -106,8 +106,8 @@ class CascadeRCNN(object):
if self.rpn_only:
im_scale = fluid.layers.slice(
im_info, [1], starts=[2], ends=[3])
im_scale = fluid.layers.sequence_expand(im_scale, rois)
rois = rois / im_scale
im_scale = fluid.layers.sequence_expand(im_scale, rpn_rois)
rois = rpn_rois / im_scale
return {'proposal': rois}
proposal_list = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册