tensorflow :: ops :: NonMaxSuppressionV4

#include <image_ops.h>

Greedily memilih subset dari kotak pembatas dalam urutan skor menurun.

Ringkasan

pemangkasan kotak yang memiliki intersection-over-union (IOU) tinggi yang tumpang tindih dengan kotak yang dipilih sebelumnya. Kotak pembatas dengan skor kurang dari score_threshold dihapus. Kotak pembatas disediakan sebagai [y1, x1, y2, x2], di mana (y1, x1) dan (y2, x2) adalah koordinat dari pasangan diagonal sudut kotak dan koordinat dapat diberikan seperti yang dinormalisasi (yaitu, terletak di interval [0, 1]) atau absolut. Perhatikan bahwa algoritma ini agnostik dimana asalnya berada dalam sistem koordinat dan lebih umum invarian terhadap transformasi ortogonal dan terjemahan dari sistem koordinat; sehingga menerjemahkan atau refleksi dari hasil sistem koordinat di kotak yang sama yang dipilih oleh algoritma. Keluaran dari operasi ini adalah sekumpulan bilangan bulat yang mengindeks ke dalam kumpulan masukan kotak pembatas yang mewakili kotak yang dipilih. Koordinat kotak pembatas yang sesuai dengan indeks yang dipilih kemudian dapat diperoleh dengan menggunakan tf.gather operation . Misalnya: selected_indices = tf.image.non_max_suppression_v2 (kotak, skor, max_output_size, iou_threshold, score_threshold) selected_boxes = tf.gather (boxes, selected_indices)

Argumen:

  • scope: Objek Scope
  • kotak: Tensor apung 2-D dengan bentuk [num_boxes, 4] .
  • skor: Tensor float 1-D bentuk [num_boxes] mewakili skor tunggal yang sesuai dengan setiap kotak (setiap baris kotak).
  • max_output_size: Tensor integer skalar yang mewakili jumlah maksimum kotak yang akan dipilih dengan penekanan non max.
  • iou_threshold: Tensor float 0-D yang mewakili ambang untuk memutuskan apakah kotak terlalu banyak tumpang tindih sehubungan dengan IOU.
  • score_threshold: Tensor float 0-D yang mewakili ambang batas untuk memutuskan kapan harus menghapus kotak berdasarkan skor.

Atribut opsional (lihat Attrs ):

  • pad_to_max_output_size: Jika true, output selected_indices akan diisi dengan panjang max_output_size . Default-nya adalah false.

Pengembalian:

  • Output selected_indices: Tensor integer 1-D bentuk [M] mewakili indeks yang dipilih dari kotak tensor, di mana M <= max_output_size .
  • Output valid_outputs: Tensor integer 0-D yang mewakili jumlah elemen valid dalam indeks_yang selected_indices , dengan elemen valid muncul pertama.

Pembuat dan Penghancur

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)

Atribut publik

operation
selected_indices
valid_outputs

Fungsi statis publik

PadToMaxOutputSize (bool x)

Structs

tensorflow :: ops :: NonMaxSuppressionV4 :: Attrs

Penyetel atribut opsional untuk NonMaxSuppressionV4 .

Atribut publik

operasi

Operation operation

selected_indices

::tensorflow::Output selected_indices

valid_outputs

::tensorflow::Output valid_outputs

Fungsi publik

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
)

Fungsi statis publik

PadToMaxOutputSize

Attrs PadToMaxOutputSize(
  bool x
)