Halaman ini diterjemahkan oleh Cloud Translation API.
Switch to English

tensorflow :: ops :: NonMaxSuppressionV2

#include <image_ops.h>

Dengan rakus memilih subset dari kotak pembatas dalam urutan skor menurun.

Ringkasan

memangkas kotak yang memiliki intersection-over-union (IOU) tinggi yang tumpang tindih dengan kotak yang dipilih sebelumnya. 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 algoritme ini bersifat agnostik karena asalnya ada di sistem koordinat. Perhatikan bahwa algoritme ini tidak berbeda dengan transformasi ortogonal dan terjemahan sistem koordinat; sehingga menerjemahkan atau refleksi dari sistem koordinat menghasilkan 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 . Sebagai contoh:

selected_indices = tf.image.non_max_suppression_v2 (kotak, skor, max_output_size, iou_threshold) selected_boxes = tf.gather (box, selected_indices)

Argumen:

  • scope: Objek Scope
  • box: Tensor float 2-D dengan bentuk [num_boxes, 4] .
  • skor: Tensor float 1-D dari 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 mewakili ambang batas untuk memutuskan apakah kotak terlalu banyak tumpang tindih sehubungan dengan IOU.

Pengembalian:

  • Output : Tensor integer 1-D bentuk [M] mewakili indeks yang dipilih dari kotak tensor, di mana M <= max_output_size .

Pembuat dan Penghancur

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

Atribut publik

operation
selected_indices

Fungsi publik

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

Atribut publik

operasi

Operation operation

selected_indices

::tensorflow::Output selected_indices

Fungsi publik

NonMaxSuppressionV2

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

simpul

::tensorflow::Node * node() const 
.dll

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Keluaran

 operator::tensorflow::Output() const