tensorflow :: ops :: NonMaxSuppressionWithOverlaps

#include <image_ops.h>

Greedily memilih subset dari kotak pembatas dalam urutan skor menurun.

Ringkasan

memangkas kotak yang memiliki tumpang tindih tinggi dengan kotak yang dipilih sebelumnya. Kotak pembatas dengan skor kurang dari score_threshold dihapus. Nilai tumpang tindih N-by-n disediakan sebagai matriks persegi, yang memungkinkan untuk menentukan kriteria tumpang tindih kustom (misalnya, persimpangan atas gabungan, persimpangan di atas area, dll.).

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_with_overlaps (tumpang tindih, skor, max_output_size, overlap_threshold, score_threshold) selected_boxes = tf.gather (box, selected_indices)

Argumen:

  • scope: Objek Scope
  • overlaps: Tensor float 2-D dengan bentuk [num_boxes, num_boxes] mewakili nilai tumpang tindih kotak n-by-n.
  • 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.
  • overlap_threshold: Tensor float 0-D mewakili ambang batas untuk memutuskan apakah kotak juga tumpang tindih.
  • score_threshold: Tensor float 0-D yang mewakili ambang batas untuk memutuskan kapan harus menghapus kotak berdasarkan skor.

Pengembalian:

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

Pembuat dan Penghancur

NonMaxSuppressionWithOverlaps (const :: tensorflow::Scope & scope, :: tensorflow::Input overlaps, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input overlap_threshold, :: tensorflow::Input score_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

NonMaxSuppressionWithOverlaps

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

simpul

::tensorflow::Node * node() const

operator :: tensorflow :: Input

 operator::tensorflow::Input() const

operator :: tensorflow :: Keluaran

 operator::tensorflow::Output() const