Kan Wu created MXNET-40:
---------------------------
Summary: add multi proposal operator (cpu version) and fix the bug
in proposal op (gpu version)
Key: MXNET-40
URL: https://issues.apache.org/jira/browse/MXNET-40
Project: Apache MXNet
Issue Type: Improvement
Reporter: Kan Wu
Assignee: Chris Olivier
The multi_proposal operator ({{mxnet.sym.contrib.MultiProposal}}, CPU version)
is not implemented before.
I wrote the code about it.
And I found there was a bug in _proposal.cu_ and _multi_proposal.cu_.
The {{batch_size}} of the output of _Proposal_ and _MultiProposal_ are both
{{param_.rpn_post_nms_top_n}}.
When {{count_anchors < param_.rpn_post_nms_top_n}}, the variable
{{rpn_post_nms_top_n}} will be {{count_anchors}}, which is less than
{{param_.rpn_post_nms_top_n}}.
[https://github.com/apache/incubator-mxnet/blob/master/src/operator/contrib/proposal.cu#L438]
It will cause the problem that the output whose index is larger than
{{rpn_post_nms_top_n}} will be _not assigned_.
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
