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

fix_generate_proposals_v2 (#5382)

上级 750ea214
......@@ -1536,6 +1536,8 @@ def generate_proposals(scores,
'pixel_offset', pixel_offset)
rpn_rois, rpn_roi_probs, rpn_rois_num = core.ops.generate_proposals_v2(
scores, bbox_deltas, im_shape, anchors, variances, *attrs)
if not return_rois_num:
rpn_rois_num = None
return rpn_rois, rpn_roi_probs, rpn_rois_num
else:
......@@ -1586,6 +1588,8 @@ def generate_proposals(scores,
outputs=outputs)
rpn_rois.stop_gradient = True
rpn_roi_probs.stop_gradient = True
if not return_rois_num:
rpn_rois_num = None
return rpn_rois, rpn_roi_probs, rpn_rois_num
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册