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 manaM <= 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