tensorflow :: ops :: CombinedNonMaxSuppression

#include <image_ops.h>

Greedily memilih subset dari kotak pembatas dalam urutan skor menurun.

Ringkasan

Operasi ini melakukan non_max_suppression pada input per batch, di semua kelas. Pangkas 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 juga 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. Output dari operasi ini adalah kotak terakhir, skor, dan tensor kelas yang dikembalikan setelah melakukan non_max_suppression.

Argumen:

  • scope: Objek Scope
  • kotak: Tensor apung 4-D dengan bentuk [batch_size, num_boxes, q, 4] . Jika q adalah 1 maka kotak yang sama digunakan untuk semua kelas jika tidak, jika q sama dengan jumlah kelas, kotak khusus kelas digunakan.
  • skor: Tensor float 3-D dari bentuk [batch_size, num_boxes, num_classes] mewakili skor tunggal yang sesuai dengan setiap kotak (setiap baris kotak).
  • max_output_size_per_class: Tensor integer skalar yang mewakili jumlah maksimum kotak yang akan dipilih dengan penekanan non maks per kelas
  • max_total_size: Skalar yang mewakili jumlah maksimum kotak yang dipertahankan di semua kelas.
  • 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.

Atribut opsional (lihat Attrs ):

  • pad_per_class: Jika salah, keluaran kotak nmsed, skor dan kelas di-padded / dipotong ke max_total_size . Jika benar, keluaran kotak nmsed, skor dan kelas akan diisi dengan panjang max_size_per_class * num_classes , kecuali jika melebihi max_total_size dalam hal ini dipotong ke max_total_size . Default-nya adalah false.
  • clip_boxes: Jika benar, asumsikan koordinat kotak berada di antara [0, 1] dan klip kotak keluaran jika melebihi [0, 1]. Jika salah, jangan melakukan pemotongan dan mengeluarkan koordinat kotak apa adanya.

Pengembalian:

  • Output nmsed_boxes: Tensor [batch_size, max_detections, 4] float32 yang berisi kotak non-max yang disembunyikan.
  • Output nmsed_scores: Tensor [batch_size, max_detections] float32 yang berisi skor untuk kotak.
  • Output nmsed_classes: Tensor [batch_size, max_detections] float32 yang berisi kelas untuk kotak.
  • Output valid_detections: Tensor [batch_size] int32 yang menunjukkan jumlah deteksi yang valid per item batch. Hanya entri num_detections [i] teratas di nms_boxes [i], nms_scores [i] dan nms_class [i] yang valid. Sisa entri lainnya adalah bantalan nol.

Pembuat dan Penghancur

CombinedNonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size_per_class, :: tensorflow::Input max_total_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold)
CombinedNonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size_per_class, :: tensorflow::Input max_total_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, const CombinedNonMaxSuppression::Attrs & attrs)

Atribut publik

nmsed_boxes
nmsed_classes
nmsed_scores
operation
valid_detections

Fungsi statis publik

ClipBoxes (bool x)
PadPerClass (bool x)

Structs

tensorflow :: ops :: CombinedNonMaxSuppression :: Attrs

Penyetel atribut opsional untuk CombinedNonMaxSuppression .

Atribut publik

nmsed_boxes

::tensorflow::Output nmsed_boxes

nmsed_classes

::tensorflow::Output nmsed_classes

nmsed_scores

::tensorflow::Output nmsed_scores

operasi

Operation operation

valid_detections

::tensorflow::Output valid_detections

Fungsi publik

CombinedNonMaxSuppression

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold
)

CombinedNonMaxSuppression

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  const CombinedNonMaxSuppression::Attrs & attrs
)

Fungsi statis publik

ClipBox

Attrs ClipBoxes(
  bool x
)

PadPerClass

Attrs PadPerClass(
  bool x
)