Halaman ini diterjemahkan oleh Cloud Translation API.
Switch to English

tensorflow :: ops :: NonMaxSuppressionV5

#include <image_ops.h>

Dengan rakus memilih subset dari kotak pembatas dalam urutan skor menurun.

Ringkasan

pemangkasan kotak yang memiliki intersection-over-union (IOU) tinggi 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 algoritme ini bersifat agnostik karena asalnya berada dalam sistem koordinat dan secara lebih umum tidak berubah dari transformasi ortogonal dan terjemahan sistem koordinat; sehingga menerjemahkan atau refleksi dari sistem koordinat menghasilkan 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 dkk , https://arxiv.org/abs/1704.04503 ) di mana kotak mengurangi skor kotak tumpang tindih lainnya 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
  • box: Tensor float 2-D dengan bentuk [num_boxes, 4] .
  • skor: Tensor float 1-D dari 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 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 bilangan bulat 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 bilangan bulat 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
)