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: issues-unsubscr...@mxnet.apache.org For additional commands, e-mail: issues-h...@mxnet.apache.org