このページは Cloud Translation API によって翻訳されました。
Switch to English

tensorflow :: ops :: CombinedNonMaxSuppression

#include <image_ops.h>

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

概要

この操作は、すべてのクラスにわたって、バッチごとの入力に対してnon_max_suppressionを実行します。交差(ユニオン)が高いボックスは、以前に選択したボックスとオーバーラップします。境界ボックスは[y1、x1、y2、x2]として提供されます。ここで、(y1、x1)と(y2、x2)はボックスコーナーの対角線のペアの座標であり、座標は正規化されて提供されます(つまり、間隔[0、1])または絶対。このアルゴリズムは、原点が座標系のどこにあるかに依存しないことに注意してください。また、このアルゴリズムは、座標系の直交変換および変換に対して不変であることに注意してください。したがって、座標系の変換または反射により、アルゴリズムによって同じボックスが選択されます。この操作の出力は、non_max_suppressionの実行後に返される最後のボックス、スコア、クラステンソルです。

引数:

  • scope: Scopeオブジェクト
  • ボックス:形状の4次元浮動小数点テンソル[batch_size, num_boxes, q, 4]qが1の場合、すべてのクラスに同じボックスが使用されます。それ以外の場合、 qがクラスの数と等しい場合、クラス固有のボックスが使用されます。
  • スコア:各ボックス(ボックスの各行)に対応する単一のスコアを表す形状[batch_size, num_boxes, num_classes]の3次元浮動小数点テンソル。
  • max_output_size_per_class:クラスごとの非最大抑制によって選択されるボックスの最大数を表すスカラー整数テンソル
  • max_total_size:すべてのクラスで保持されるボックスの最大数を表すスカラー。
  • iou_threshold:IOUに関してボックスがオーバーラップしすぎるかどうかを決定するためのしきい値を表す0-D浮動小数点テンソル。
  • score_threshold:スコアに基づいてボックスをいつ削除するかを決定するためのしきい値を表す0-D floatテンソル。

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

  • pad_per_class:falseの場合、出力nmsedボックス、スコア、およびクラスはmax_total_sizeパディング/クリップされmax_total_size 。 trueの場合、出力nmsedボックス、スコアおよびクラスは長さであることが埋め込まれmax_size_per_class * num_classesそれを超えない限り、 max_total_sizeクリッピングされる場合にmax_total_size 。デフォルトはfalseです。
  • clip_boxes:trueの場合、ボックスの座標が[0、1]の間にあると想定し、出力ボックスが[0、1]を超える場合はクリップします。 falseの場合、クリッピングを行わず、ボックス座標をそのまま出力します。

戻り値:

  • Output nmsed_boxes:非最大抑制ボックスを含む[batch_size、max_detections、4] float32テンソル。
  • Output nmsed_scores:ボックスのスコアを含む[batch_size、max_detections] float32テンソル。
  • Output nmsed_classes:ボックスのクラスを含む[batch_size、max_detections] float32テンソル。
  • Output valid_detections:バッチアイテムごとの有効な検出数を示す[batch_size] int32テンソル。 nms_boxes [i]、nms_scores [i]、およびnms_class [i]の上位のnum_detections [i]エントリのみが有効です。残りのエントリはゼロパディングです。

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

CombinedNonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size_per_class, :: tensorflow::Input max_total_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold)
CombinedNonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size_per_class, :: tensorflow::Input max_total_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, const CombinedNonMaxSuppression::Attrs & attrs)

公開属性

nmsed_boxes
nmsed_classes
nmsed_scores
operation
valid_detections

パブリック静的関数

ClipBoxes (bool x)
PadPerClass (bool x)

構造

tensorflow :: ops :: CombinedNonMaxSuppression :: Attrs

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

公開属性

nmsed_boxes

::tensorflow::Output nmsed_boxes

nmsed_classes

::tensorflow::Output nmsed_classes

nmsed_scores

::tensorflow::Output nmsed_scores

操作

Operation operation

valid_detections

::tensorflow::Output valid_detections

公開機能

CombinedNonMaxSuppression

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold
)

CombinedNonMaxSuppression

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  const CombinedNonMaxSuppression::Attrs & attrs
)

パブリック静的関数

クリップボックス

Attrs ClipBoxes(
  bool x
)

PadPerClass

Attrs PadPerClass(
  bool x
)