tensorflow :: ops :: NonMaxSuppressionV5

#include <image_ops.h>

Greedily memilih subset dari kotak pembatas dalam urutan skor menurun.

Ringkasan

pemangkasan kotak yang memiliki intersection-over-union (IOU) tinggi yang tumpang tindih dengan kotak yang dipilih sebelumnya. Kotak pembatas dengan skor kurang dari score_threshold dihapus. 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 algoritma ini agnostik dimana asalnya berada dalam sistem koordinat dan lebih umum invarian terhadap transformasi ortogonal dan terjemahan dari 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_v2 (boxes, score, max_output_size, iou_threshold, score_threshold) selected_boxes = tf.gather (boxes, selected_indices) Operasi ini juga mendukung mode Soft-NMS (dengan bobot Gaussian) (cf Bodla et al , https://arxiv.org/abs/1704.04503 ) di mana kotak mengurangi skor kotak lain yang tumpang tindih alih-alih secara langsung menyebabkannya dipangkas. Untuk mengaktifkan mode Soft-NMS ini, setel parameter soft_nms_sigma menjadi lebih besar dari 0.

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.
  • iou_threshold: Tensor float 0-D yang mewakili ambang untuk memutuskan apakah kotak terlalu banyak tumpang tindih sehubungan dengan IOU.
  • score_threshold: Tensor float 0-D yang mewakili ambang batas untuk memutuskan kapan harus menghapus kotak berdasarkan skor.
  • soft_nms_sigma: Tensor float 0-D yang mewakili parameter sigma untuk Soft NMS; lihat Bodla dkk (lihat https://arxiv.org/abs/1704.04503 ). Ketika soft_nms_sigma=0.0 (yang merupakan default), kita kembali ke NMS standar (hard).

Atribut opsional (lihat Attrs ):

  • pad_to_max_output_size: Jika true, output selected_indices akan diisi dengan panjang max_output_size . Default-nya adalah false.

Pengembalian:

  • Output selected_indices: Tensor integer 1-D bentuk [M] mewakili indeks yang dipilih dari kotak tensor, di mana M <= max_output_size .
  • Output selected_scores: Tensor float 1-D bentuk [M] mewakili skor yang sesuai untuk setiap kotak yang dipilih, di mana M <= max_output_size . Skor hanya berbeda dari skor input yang sesuai saat menggunakan Soft NMS (yaitu saat soft_nms_sigma>0 )
  • Output valid_outputs: Tensor integer 0-D yang mewakili jumlah elemen valid dalam indeks_yang selected_indices , dengan elemen valid muncul pertama.

Pembuat dan Penghancur

NonMaxSuppressionV5 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, :: tensorflow::Input soft_nms_sigma)
NonMaxSuppressionV5 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, :: tensorflow::Input soft_nms_sigma, const NonMaxSuppressionV5::Attrs & attrs)

Atribut publik

operation
selected_indices
selected_scores
valid_outputs

Fungsi statis publik

PadToMaxOutputSize (bool x)

Structs

tensorflow :: ops :: NonMaxSuppressionV5 :: Attrs

Penyetel atribut opsional untuk NonMaxSuppressionV5 .

Atribut publik

operasi

Operation operation

selected_indices

::tensorflow::Output selected_indices

selected_scores

::tensorflow::Output selected_scores

valid_outputs

::tensorflow::Output valid_outputs

Fungsi publik

NonMaxSuppressionV5

 NonMaxSuppressionV5(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  ::tensorflow::Input soft_nms_sigma
)

NonMaxSuppressionV5

 NonMaxSuppressionV5(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  ::tensorflow::Input soft_nms_sigma,
  const NonMaxSuppressionV5::Attrs & attrs
)

Fungsi statis publik

PadToMaxOutputSize

Attrs PadToMaxOutputSize(
  bool x
)