ExperimentalDetectronTopKROIs

バージョン名: ExperimentalDetectronTopKROIs-6

カテゴリー: ソートと最大化

簡単な説明: ExperimentalDetectronTopKROIs 操作は、入力 ROI の確率に適用される TopK 操作です。

詳細な説明: この操作では、入力 ROI に対して確率降順の並べ替えが実行され、ROI の max_rois 数が返されます。等しい確率でソートされた ROI の順序は未定義です。ROI 数が max_rois より小さい場合、操作はすべての ROI を降順ソートして返し、出力テンソルには残りの出力テンソル要素の未定義の値が埋められます。

属性:

• max_rois

  • 説明: max_rois 属性は、出力 ROI の最大数を指定します。

  • 値の範囲: 負でない整数

  • タイプ: int

  • デフォルト値: 0

  • 必須: いいえ

入力:

• 1: ROI を 4 つのタプル [x 1, y 1, x 2, y 2] として記述する、形状 [number_of_ROIs, 4] のタイプ T の 2D テンソル。必須。

• 2: 形状 [number_of_input_ROIs] を持つタイプ T の 1D テンソルには、入力 ROI の確率が含まれます。必須。

出力:

• 1: 最も高い確率で max_rois ROI を記述する、形状 [max_rois, 4] を持つタイプ T の 2D テンソル。

タイプ:

• T: サポートされている浮動小数点タイプ。

例:

<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>