tensorflow:: ops:: NonMaxSuppressionWithOverlaps

スコアの降順でバウンディングボックスのサブセットを貪欲に選択します。

まとめ

以前に選択されたボックスと重なりが大きいボックスを削除します。スコアがscore_threshold未満のバウンディングボックスは削除されます。N×Nの重なり値は正方行列として指定され、カスタムの重なり基準（例：交差と和集合、交差と面積など）を定義できます。

この操作の出力は、選択されたボックスを表す入力バウンディングボックスのコレクションへのインデックスとなる整数のセットです。選択されたインデックスに対応するバウンディングボックスの座標は、 tf.gather operationを使用して取得できます。例:

selected_indices = tf.image.non_max_suppression_with_overlaps( overlaps, scores, max_output_size, overlap_threshold, score_threshold) selected_boxes = tf.gather(boxes, selected_indices)

議論:

• スコープ:スコープオブジェクト
• overlaps: n×nボックスの重なり値を表す形状[num_boxes, num_boxes]の2次元浮動小数点テンソル。
• scores: 各ボックス（ボックスの各行）に対応する単一のスコアを表す、形状[num_boxes]の1次元浮動小数点テンソル。
• max_output_size: 最大抑制以外の方法で選択されるボックスの最大数を表すスカラー整数テンソル。
• overlap_threshold: ボックスが重なっているかどうかを判断するための閾値を表す 0 次元の浮動小数点テンソル。
• score_threshold: スコアに基づいてボックスを削除するタイミングを決定するためのしきい値を表す 0 次元の float テンソル。

返品:

• Output : ボックステンソルから選択されたインデックスを表す形状[M]の 1 次元整数テンソル。ただし、 M <= max_output_sizeです。

公開属性

Operation operation

::tensorflow::Output selected_indices

 NonMaxSuppressionWithOverlaps(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input overlaps,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input overlap_threshold,
  ::tensorflow::Input score_threshold
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const