提交 1aca439c 编写于 作者: keineahnung2345's avatar keineahnung2345 提交者: Waleed

Fix the input shape of anchors in ProposalLayer (#961)

Fix the input shape of anchors in ProposalLayer and use "num_anchors" to replace "anchors" to avoid confusing
上级 d16c4635
......@@ -258,9 +258,9 @@ class ProposalLayer(KE.Layer):
box refinement deltas to anchors.
Inputs:
rpn_probs: [batch, anchors, (bg prob, fg prob)]
rpn_bbox: [batch, anchors, (dy, dx, log(dh), log(dw))]
anchors: [batch, (y1, x1, y2, x2)] anchors in normalized coordinates
rpn_probs: [batch, num_anchors, (bg prob, fg prob)]
rpn_bbox: [batch, num_anchors, (dy, dx, log(dh), log(dw))]
anchors: [batch, num_anchors, (y1, x1, y2, x2)] anchors in normalized coordinates
Returns:
Proposals in normalized coordinates [batch, rois, (y1, x1, y2, x2)]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册