tensorflow::ops::NonMaxSuppressionWithOverlaps

Memilih secara rakus sejumlah subset kotak pembatas dalam urutan menurun berdasarkan skor.

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-by-n diberikan sebagai matriks persegi, yang memungkinkan untuk mendefinisikan kriteria tumpang tindih khusus (misalnya, irisan dibagi gabungan, irisan dibagi luas, dll.).

Output dari operasi ini adalah sekumpulan bilangan bulat yang mengindeks koleksi input kotak pembatas yang mewakili kotak yang dipilih. Koordinat kotak pembatas yang sesuai dengan indeks yang dipilih kemudian dapat diperoleh menggunakan tf.gather operation . Misalnya:

indeks_terpilih = tf.image.non_max_suppression_with_overlaps( overlaps, scores, max_output_size, overlap_threshold, score_threshold) kotak_terpilih = tf.gather(boxes, indeks_terpilih)

Argumen:

• Cakupan: Objek Cakupan
• overlaps: Tensor float 2-D dengan bentuk [num_boxes, num_boxes] yang merepresentasikan nilai tumpang tindih kotak n-by-n.
• skor: Tensor float 1-D dengan bentuk [num_boxes] yang mewakili satu skor yang sesuai dengan setiap kotak (setiap baris kotak).
• max_output_size: Sebuah tensor bilangan bulat skalar yang mewakili jumlah maksimum kotak yang akan dipilih oleh penekanan non-maksimum.
• overlap_threshold: Tensor float 0-D yang mewakili ambang batas untuk menentukan apakah kotak-kotak tersebut saling tumpang tindih.
• score_threshold: Tensor float 0-D yang mewakili ambang batas untuk memutuskan kapan harus menghapus kotak berdasarkan skor.

Mengembalikan:

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

Atribut publik

Operation operation

::tensorflow::Output selected_indices

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const