Skip to content

P
PaddleDetection

PaddlePaddle / PaddleDetection
大约 1 年 前同步成功

通知 695
Star 11112
Fork 2696
P
PaddleDetection
提交 3bc5b56b 编写于 作者: W wangguanzhong 提交者： GitHub
浏览文件
操作

fix rpn_only in cascade mask fpn (#3231) 

* fix rpn_only in cascade mask fpn

* fix rpn_only in cascade rcnn
上级 14c5f645
隐藏空白更改
内联 并排
Showing with 4 addition and 4 deletion
ppdet/modeling/architectures/cascade_mask_rcnn.py
浏览文件 @ 3bc5b56b
...@@ -114,8 +114,8 @@ class CascadeMaskRCNN(object): ...@@ -114,8 +114,8 @@ class CascadeMaskRCNN(object):
if self.rpn_only: if self.rpn_only:
im_scale = fluid.layers.slice( im_scale = fluid.layers.slice(
im_info, [1], starts=[2], ends=[3]) im_info, [1], starts=[2], ends=[3])
im_scale = fluid.layers.sequence_expand(im_scale, rois) im_scale = fluid.layers.sequence_expand(im_scale, rpn_rois)
rois = rois / im_scale rois = rpn_rois / im_scale
return {'proposal': rois} return {'proposal': rois}
proposal_list = [] proposal_list = []
......
ppdet/modeling/architectures/cascade_rcnn.py
浏览文件 @ 3bc5b56b
...@@ -106,8 +106,8 @@ class CascadeRCNN(object): ...@@ -106,8 +106,8 @@ class CascadeRCNN(object):
if self.rpn_only: if self.rpn_only:
im_scale = fluid.layers.slice( im_scale = fluid.layers.slice(
im_info, [1], starts=[2], ends=[3]) im_info, [1], starts=[2], ends=[3])
im_scale = fluid.layers.sequence_expand(im_scale, rois) im_scale = fluid.layers.sequence_expand(im_scale, rpn_rois)
rois = rois / im_scale rois = rpn_rois / im_scale
return {'proposal': rois} return {'proposal': rois}
proposal_list = [] proposal_list = []
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或