tensorflow :: ops :: NonMaxSuppressionV2

#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 precedentemente selezionate. I riquadri di selezione 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 l'intervallo [0, 1]) o assoluto. Notare che questo algoritmo è agnostico rispetto alla posizione dell'origine 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 delimitazione che rappresentano i riquadri selezionati. Le coordinate del riquadro di delimitazione corrispondenti agli indici selezionati possono quindi essere ottenute utilizzando l' tf.gather operation . Per esempio:

selected_indices = tf.image.non_max_suppression_v2 (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.
  • iou_threshold: un tensore a virgola mobile 0-D 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

NonMaxSuppressionV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold)

Attributi pubblici

operation
selected_indices

Funzioni pubbliche

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

Attributi pubblici

operazione

Operation operation

selected_indices

::tensorflow::Output selected_indices

Funzioni pubbliche

NonMaxSuppressionV2

 NonMaxSuppressionV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold
)

nodo

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const