テンソルフロー::作戦::オーバーラップありの非最大抑制

#include <image_ops.h>

スコアの降順で境界ボックスのサブセットを貪欲に選択します。

まとめ

以前に選択したボックスと重複する部分が多いボックスを削除します。スコアがscore_threshold未満の境界ボックスは削除されます。 N 行 n 列のオーバーラップ値は正方行列として提供され、カスタムのオーバーラップ基準 (和集合上の交差、エリア上の交差など) を定義できます。

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

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

引数:

  • スコープ:スコープオブジェクト
  • オーバーラップ: n 行 n 列のボックスのオーバーラップ値を表す形状[num_boxes, num_boxes]の 2 次元浮動小数点テンソル。
  • スコア: 各ボックス (ボックスの各行) に対応する単一のスコアを表す、形状[num_boxes]の 1 次元浮動小数点テンソル。
  • max_output_size: 非最大抑制によって選択されるボックスの最大数を表すスカラー整数テンソル。
  • overlap_threshold: ボックスもオーバーラップするかどうかを決定するためのしきい値を表す 0 次元浮動小数点テンソル。
  • core_threshold: スコアに基づいてボックスをいつ削除するかを決定するためのしきい値を表す 0 次元の浮動小数点テンソル。

戻り値:

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

コンストラクターとデストラクター

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

パブリック属性

operation
selected_indices

公共機能

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

パブリック属性

手術

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 

演算子::tensorflow::入力

 operator::tensorflow::Input() const 

演算子::tensorflow::出力

 operator::tensorflow::Output() const