tensorflow :: ops :: NonMaxSuppression
#include <image_ops.h>
Greedily 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 agnostik karena asalnya ada di sistem koordinat. Perhatikan bahwa algoritme ini berbeda dengan transformasi ortogonal dan terjemahan sistem koordinat; sehingga menerjemahkan atau refleksi dari hasil sistem koordinat di 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
. Misalnya: selected_indices = tf.image.non_max_suppression (kotak, skor, max_output_size, iou_threshold) selected_boxes = tf.gather (boxes, selected_indices)
Argumen:
- scope: Objek Scope
- kotak: Tensor apung 2-D dengan bentuk
[num_boxes, 4]
. - 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.
Atribut opsional (lihat Attrs
):
- iou_threshold: Float yang mewakili ambang 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 manaM <= max_output_size
.
Pembuat dan Penghancur | |
---|---|
NonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size) | |
NonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, const NonMaxSuppression::Attrs & attrs) |
Atribut publik | |
---|---|
operation | |
selected_indices |
Fungsi publik | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Fungsi statis publik | |
---|---|
IouThreshold (float x) |
Structs | |
---|---|
tensorflow :: ops :: NonMaxSuppression :: Attrs | Penyetel atribut opsional untuk NonMaxSuppression . |
Atribut publik
operasi
Operation operation
selected_indices
::tensorflow::Output selected_indices
Fungsi publik
NonMaxSuppression
NonMaxSuppression( const ::tensorflow::Scope & scope, ::tensorflow::Input boxes, ::tensorflow::Input scores, ::tensorflow::Input max_output_size )
NonMaxSuppression
NonMaxSuppression( const ::tensorflow::Scope & scope, ::tensorflow::Input boxes, ::tensorflow::Input scores, ::tensorflow::Input max_output_size, const NonMaxSuppression::Attrs & attrs )
simpul
::tensorflow::Node * node() const
operator :: tensorflow :: Input
operator::tensorflow::Input() const
operator :: tensorflow :: Keluaran
operator::tensorflow::Output() const
Fungsi statis publik
IouThreshold
Attrs IouThreshold( float x )