aliran tensor:: operasi:: NonMaxSuppressionDengan Tumpang Tindih

#include <image_ops.h>

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

Ringkasan

memangkas kotak-kotak yang memiliki tumpang tindih tinggi dengan kotak-kotak yang dipilih sebelumnya. Kotak pembatas dengan skor kurang dari score_threshold akan dihapus. Nilai tumpang tindih N-kali-n diberikan dalam bentuk matriks persegi, yang memungkinkan untuk menentukan kriteria tumpang tindih khusus (misalnya perpotongan dengan kesatuan, perpotongan dengan luas, dll.).

Output dari operasi ini adalah sekumpulan bilangan bulat yang diindeks ke dalam kumpulan input 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:

indeks_yang dipilih = tf.image.non_max_suppression_with_overlaps( tumpang tindih, skor, ukuran_output_maks, ambang_tumpang tindih, ambang_skor) kotak_yang dipilih = tf.gather(kotak, indeks_yang dipilih)

Argumen:

  • ruang lingkup: Objek Lingkup
  • tumpang tindih: Tensor float 2-D dengan bentuk [num_boxes, num_boxes] yang mewakili nilai tumpang tindih kotak n-kali-n.
  • skor: Tensor float 1-D berbentuk [num_boxes] yang mewakili satu skor yang sesuai dengan setiap kotak (setiap baris kotak).
  • max_output_size: Tensor bilangan bulat skalar yang mewakili jumlah maksimum kotak yang akan dipilih dengan penekanan non-maks.
  • overlap_threshold: Tensor float 0-D yang mewakili ambang batas untuk menentukan 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 bilangan bulat 1-D berbentuk [M] yang mewakili indeks yang dipilih dari kotak tensor, dengan M <= max_output_size .

Konstruktor dan Destruktor

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

indeks_yang dipilih

::tensorflow::Output selected_indices

Fungsi publik

NonMaxSuppressionDengan Tumpang Tindih

 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::Masukan

 operator::tensorflow::Input() const 

operator::tensorflow::Keluaran

 operator::tensorflow::Output() const