tensorflow :: ops :: NonMaxSuppression

#include <image_ops.h>

Seleziona avidamente un sottoinsieme di riquadri di delimitazione in ordine decrescente di punteggio.

Sommario

eliminazione di scatole che hanno un'elevata sovrapposizione di intersezione su unione (IOU) con scatole selezionate in precedenza. I riquadri di delimitazione sono forniti come [y1, x1, y2, x2], dove (y1, x1) e (y2, x2) sono le coordinate di qualsiasi coppia diagonale di angoli del riquadro e le coordinate possono essere fornite come normalizzate (cioè, giacenti in l'intervallo [0, 1]) o assoluto. Notare che questo algoritmo è agnostico rispetto al punto in cui l'origine si trova nel sistema di coordinate. Notare che questo algoritmo è invariante alle trasformazioni ortogonali e alle traslazioni del sistema di coordinate; quindi la traduzione o le riflessioni del sistema di coordinate risultano nelle stesse caselle selezionate dall'algoritmo. L'output di questa operazione è un insieme di numeri interi indicizzati nella raccolta di input di riquadri di selezione che rappresentano i riquadri selezionati. Le coordinate del riquadro di delimitazione corrispondenti agli indici selezionati possono quindi essere ottenute utilizzando l' tf.gather operation . Ad esempio: selected_indices = tf.image.non_max_suppression (box, score, max_output_size, iou_threshold) selected_boxes = tf.gather (box, selected_indices)

Argomenti:

  • scope: un oggetto Scope
  • riquadri: un tensore a virgola mobile 2-D di forma [num_boxes, 4] .
  • punteggi: un tensore in virgola mobile 1-D di forma [num_boxes] rappresenta un singolo punteggio corrispondente a ciascuna casella (ciascuna riga di caselle).
  • max_output_size: un tensore intero scalare che rappresenta il numero massimo di caselle che possono essere selezionate dalla soppressione non max.

Attributi opzionali (vedi Attrs ):

  • iou_threshold: un float che rappresenta la soglia per decidere se le caselle si sovrappongono troppo rispetto a IOU.

Ritorna:

  • Output : Un tensore intero 1-D di forma [M] rappresenta gli indici selezionati dal tensore delle caselle, dove M <= max_output_size .

Costruttori e distruttori

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)

Attributi pubblici

operation
selected_indices

Funzioni pubbliche

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Funzioni statiche pubbliche

IouThreshold (float x)

Structs

tensorflow :: ops :: NonMaxSuppression :: Attrs

Setter di attributi facoltativi per NonMaxSuppression .

Attributi pubblici

operazione

Operation operation

selected_indices

::tensorflow::Output selected_indices

Funzioni pubbliche

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
)

nodo

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const 

Funzioni statiche pubbliche

IouThreshold

Attrs IouThreshold(
  float x
)