BoundingBoxProposal の生成

パブリック最終クラスGenerateBoundingBoxProposals

この操作は、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
create ( Scopeスコープ、オペランド<Float> スコア、オペランド<Float> bboxDeltas、オペランド<Float> imageInfo、オペランド<Float> アンカー、オペランド<Float> nmsThreshold、オペランド<Integer> preNmsTopn、オペランド<Float> minSize、 Options.. .オプション)
新しい 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 次元浮動小数点数テンソル。スコアの降順に並べ替えられます。