[object Object][object Object][object Object]undefined
[object Object]
  • Description: Performs backpropagation of RoiPoolingWithArgMax. Traverse the pooling result of each ROI and accumulate the backward gradient contributions on the coordinates of the feature map, that is, complete the backward propagation on the entire image.
  • Formulas:Lxi=rj[i=i(r,j)]Lyrj\frac{\partial L}{\partial x_i} = \sum_{r}\sum_{j}[i = i^*(r,j)]\frac{\partial L}{\partial y_{rj}} Wherein:[i=i(r,j)]={1,i(r,j)10,otherwise[i = i^*(r,j)] = \begin{cases} 1, & i^*(r,j) \geq 1 \\ 0, & otherwise \end{cases} The decision function [i = i^*(r,j)] indicates whether the node i is selected as the maximum value output by the jth output node of the candidate region r.
[object Object]

Each operator has calls. First, [object Object] is called to obtain the input parameters and compute the required workspace size based on the process. Then, [object Object] is called to perform computation.

[object Object]
[object Object]
[object Object]
  • Parameters

    [object Object]
  • Returns:

    [object Object]: status code. For details, see .

    The first-phase API implements input parameter validation. The following error codes may be returned.

    [object Object]
[object Object]
  • Parameters:

    [object Object]
  • Return values:

    [object Object]: status code. For details, see .

[object Object]
  1. The data types of gradOutput, rois, argmax, and gradInputRef are within the supported range.
  2. gradOutput, argmax, and gradInputRef have the same data type.
  3. The shape of gradOutput, argmax, and gradInputRef is 4, and the shape of rois is 2.
  4. The shape[0] of gradOutput, argmax, and rois is the same.
  5. The shape[1] of gradOutput and argmax is the same.
  6. The shape[2] of gradOutput and argmax is equal to pooledH and shape[3] is equal to pooledW.
  7. The value of rois is greater than or equal to 0.
  8. pooledH and pooledW are greater than 0.
  9. rois[:, 1] is less than rois[:, 2] and rois[:, 3] is less than rois[:, 4].
  10. rois.shape[0] and gradOutput.shape[0] are less than or equal to 1024.
[object Object]

The following example is for reference only. For details, see .

[object Object]