この操作は、arXiv:1506.01497 の eq.2 に従って、指定された境界ボックス (bbox_deltas) でエンコードされた wrt アンカーから関心領域を生成します。
この操作は、上位の「pre_nms_topn」スコアリングボックスを選択し、アンカーに関してそれらをデコードし、「nms_threshold」intersection-over-union (iou) 値よりも高い重なり合うボックスに非最大抑制を適用し、短い辺が ` より小さいボックスを破棄します。 min_size`。入力: `scores`: 指定された位置のアンカーごとのスコアを含む形状 [バッチ、高さ、幅、アンカー数] の 4D テンソル `bbox_deltas`: 形状 [バッチ、高さ、幅、4 x アンカー数] のテンソルです各アンカーにエンコードされたボックス `anchors`: アンカーを表す、形状 [4 x Num Anchors] の 1D テンソル。出力: `rois`: 出力 RoIs、形状 [Batch, post_nms_topn, 4] の 3D テンソル。見つかった post_nms_topn 候補よりも小さい場合は 0 がパディングされます。 `roi_probabilities`: 形状 [Batch,post_nms_topn] の 2D テンソルである「rois」の各 roi の確率スコア。必要に応じて 0 が埋め込まれ、スコアによってソートされます。
ネストされたクラス
クラス | GenerateBoundingBoxProposals.Options | GenerateBoundingBoxProposals のオプションの属性 |
パブリックメソッド
静的GenerateBoundingBoxProposals | |
静的GenerateBoundingBoxProposals.Options | postNmsTopn (ロングポストNmsTopn) |
出力<浮動小数点数> | roi確率() 同じインデックスの rois テンソルの関心領域ボックスのスコアを表す、形状 `[num_images, post_nms_topn]` の 2 次元浮動小数点テンソル。 |
出力<浮動小数点数> | ロイス() 選択された関心領域ボックスを表す形状 `[num_images,post_nms_topn,4]` の 3 次元浮動小数点数テンソル。 |
継承されたメソッド
パブリックメソッド
public static GenerateBoundingBoxProposals create ( Scopeスコープ、オペランド<Float> スコア、オペランド<Float> bboxDeltas、オペランド<Float> imageInfo、オペランド<Float> アンカー、オペランド<Float> nmsThreshold、オペランド<Integer> preNmsTopn、オペランド<Float> minSize、オプション...オプション)
新しい GenerateBoundingBoxProposals 操作をラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
スコア | 指定されたアンカーのボックスのスコアを含む形状 `[num_images, height, width, num_achors]` の 4 次元浮動小数点数テンソルは、ソート解除できます。 |
bboxデルタ | 形状 `[num_images, height, width, 4 x num_anchors]` の 4 次元浮動小数点テンソル。各アンカーに関してボックスをエンコードします。座標は [dy, dx, dh, dw] の形式で指定されます。 |
画像情報 | 画像情報の高さ、幅、スケールを含む形状 `[num_images, 5]` の 2 次元浮動小数点テンソル。 |
アンカー | アンカー ボックスを記述する形状 `[num_anchors, 4]` の 2 次元浮動小数点テンソル。ボックスは [y1, x1, y2, x2] の形式でフォーマットされます。 |
nmsしきい値 | 非極大抑制しきい値のスカラー浮動小数点数テンソル。 |
preNmsTopn | 入力として使用される最高得点ボックスの数のスカラー int テンソル。 |
最小サイズ | スカラー浮動小数点テンソル。 min_size より小さいサイズのボックスはすべて破棄されます。 |
オプション | オプションの属性値を持ちます |
戻り値
- GenerateBoundingBoxProposals の新しいインスタンス
public static GenerateBoundingBoxProposals.Options postNmsTopn (ロング postNmsTopn)
パラメーター
postNmsTopn | 整数。出力内の rois の最大数。 |
---|
public出力<Float> roiProbabilities ()
同じインデックスの rois テンソルの関心領域ボックスのスコアを表す、形状 `[num_images, post_nms_topn]` の 2 次元浮動小数点テンソル。
public出力<Float> rois ()
選択された関心領域ボックスを表す形状 `[num_images,post_nms_topn,4]` の 3 次元浮動小数点数テンソル。スコアの降順に並べ替えられます。