Details
-
Improvement
-
Status: To Do
-
Major
-
Resolution: Unresolved
-
None
-
None
Description
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.
Attachments
Issue Links
- links to