tensorflow::ops::NonMaxSuppression

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

Ringkasan

Menghilangkan kotak-kotak yang memiliki tumpang tindih intersection-over-union (IOU) yang tinggi dengan kotak-kotak yang telah dipilih sebelumnya. Kotak pembatas diberikan sebagai [y1, x1, y2, x2], di mana (y1, x1) dan (y2, x2) adalah koordinat dari setiap pasangan diagonal sudut kotak dan koordinat dapat diberikan sebagai koordinat yang dinormalisasi (yaitu, berada dalam interval [0, 1]) atau absolut. Perhatikan bahwa algoritma ini tidak bergantung pada letak titik asal dalam sistem koordinat. Perhatikan bahwa algoritma ini invarian terhadap transformasi ortogonal dan translasi sistem koordinat; sehingga translasi atau refleksi sistem koordinat menghasilkan kotak yang sama yang dipilih oleh algoritma. Keluaran dari operasi ini adalah sekumpulan bilangan bulat yang mengindeks ke dalam 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 . Contohnya: selected_indices = tf.image.non_max_suppression( boxes, scores, max_output_size, iou_threshold) selected_boxes = tf.gather(boxes, selected_indices)

Argumen:

• Cakupan: Objek Cakupan
• boxes: Tensor float 2-D dengan bentuk [num_boxes, 4] .
• 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.

Atribut opsional (lihat Attrs ):

• iou_threshold: Nilai float yang mewakili ambang batas untuk menentukan apakah kotak-kotak tersebut terlalu tumpang tindih terkait dengan IOU.

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

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs IouThreshold(
  float x
)