ExperimentalDetectronTopKROIs

Versioned name: ExperimentalDetectronTopKROIs-6

Category: Sort

Short description: The ExperimentalDetectronTopKROIs operation is TopK operation applied to probabilities of input ROIs.

Detailed description: The operation performs probabilities descending sorting for input ROIs and returns max_rois number of ROIs. Order of sorted ROIs with equal probabilities is undefined. If the number of ROIs is less than max_rois then operation returns all ROIs descended sorted and the output tensor is filled with undefined values for the rest of output tensor elements.

Attributes:

• max_rois
  • Description: The max_rois attribute specifies maximal numbers of output ROIs.
  • Range of values: non-negative integer number
  • Type: int
  • Default value: 0
  • Required: no

Inputs

• 1: A 2D tensor of type T with shape [number_of_ROIs, 4] describing the ROIs as 4-tuples: [x₁, y₁, x₂, y₂]. Required.
• 2: A 1D tensor of type T with shape [number_of_input_ROIs] contains probabilities for input ROIs. Required.

Outputs

• 1: A 2D tensor of type T with shape [max_rois, 4] describing max_rois ROIs with highest probabilities.

Types

• T: any supported floating point type.

Example

<layer ... type="ExperimentalDetectronTopKROIs" version="opset6">
    <data max_rois="1000"/>
    <input>
        <port id="0">
            <dim>5000</dim>
            <dim>4</dim>
        </port>
        <port id="1">
            <dim>5000</dim>
        </port>
    </input>
    <output>
        <port id="2" precision="FP32">
            <dim>1000</dim>
            <dim>4</dim>
        </port>
    </output>
</layer>