tensorflow :: ops :: NonMaxSuppressionV4

#include <image_ops.h>

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

概要

高い交差オーバーユニオン(IOU)を持つボックスを剪定すると、以前に選択したボックスと重複します。スコアがscore_threshold未満の境界ボックスは削除されます。境界ボックスは[y1、x1、y2、x2]として提供されます。ここで、(y1、x1)および(y2、x2)はボックスの角の対角線のペアの座標であり、座標は正規化されたものとして提供できます(つまり、間隔[0、1])または絶対。このアルゴリズムは、原点が座標系のどこにあるかに依存せず、より一般的には、座標系の直交変換および平行移動に対して不変であることに注意してください。したがって、座標系の変換または反射により、アルゴリズムによって同じボックスが選択されます。この操作の出力は、選択されたボックスを表すバウンディングボックスの入力コレクションにインデックスを付ける整数のセットです。選択したインデックスに対応するバウンディングボックスの座標は、 tf.gather operationを使用して取得できtf.gather operation 。例:selected_indices = tf.image.non_max_suppression_v2(boxes、scores、max_output_size、iou_threshold、score_threshold)selected_boxes = tf.gather(boxes、selected_indices)

引数:

  • スコープ:スコープオブジェクト
  • ボックス:形状[num_boxes, 4] 2次元浮動小数点テンソル。
  • スコア:各ボックス(ボックスの各行)に対応する単一のスコアを表す形状[num_boxes]の1次元浮動小数点テンソル。
  • max_output_size:非最大抑制によって選択されるボックスの最大数を表すスカラー整数テンソル。
  • iou_threshold:ボックスがIOUに対してオーバーラップしすぎているかどうかを判断するためのしきい値を表す0-D浮動小数点テンソル。
  • score_threshold:スコアに基づいてボックスを削除するタイミングを決定するためのしきい値を表す0-D浮動小数点テンソル。

オプションの属性(参照Attrs ):

  • pad_to_max_output_size:trueの場合、出力selected_indicesは、長さがmax_output_sizeになるようにパディングされます。デフォルトはfalseです。

戻り値:

  • Output selected_indices:ボックステンソルから選択されたインデックスを表す形状[M] 1次元整数テンソル。ここで、 M <= max_output_size
  • Output valid_outputs:で有効な要素の数を表すテンソル整数0-D selected_indices 、最初に出現する有効な要素を有します。

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

NonMaxSuppressionV4 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold)
NonMaxSuppressionV4 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, const NonMaxSuppressionV4::Attrs & attrs)

パブリック属性

operation
selected_indices
valid_outputs

パブリック静的関数

PadToMaxOutputSize (bool x)

構造体

tensorflow :: ops :: NonMaxSuppressionV4 :: Attrs

NonMaxSuppressionV4のオプションの属性セッター。

パブリック属性

操作

Operation operation

selected_indices

::tensorflow::Output selected_indices

valid_outputs

::tensorflow::Output valid_outputs

公の行事

NonMaxSuppressionV4

 NonMaxSuppressionV4(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold
)

NonMaxSuppressionV4

 NonMaxSuppressionV4(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  const NonMaxSuppressionV4::Attrs & attrs
)

パブリック静的関数

PadToMaxOutputSize

Attrs PadToMaxOutputSize(
  bool x
)