Uploaded image for project: 'Apache MXNet (Retired)'
  1. Apache MXNet (Retired)
  2. MXNET-40

add multi proposal operator (cpu version) and fix the bug in proposal op (gpu version)

    XMLWordPrintableJSON

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

          Activity

            People

              Unassigned Unassigned
              wkcn Jackie Wu
              Votes:
              0 Vote for this issue
              Watchers:
              2 Start watching this issue

              Dates

                Created:
                Updated:

                Time Tracking

                  Estimated:
                  Original Estimate - Not Specified
                  Not Specified
                  Remaining:
                  Remaining Estimate - 0h
                  0h
                  Logged:
                  Time Spent - 2.5h
                  2.5h