tensoreflusso:: ops:: NonMaxSuppression
#include <image_ops.h>
Seleziona avidamente un sottoinsieme di riquadri di delimitazione in ordine decrescente di punteggio.
Riepilogo
eliminando le caselle che hanno un'elevata sovrapposizione di intersezione su unione (IOU) con le caselle precedentemente selezionate. I riquadri di delimitazione vengono 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 (ovvero, giacenti in l'intervallo [0, 1]) o assoluto. Si noti che questo algoritmo è indipendente dalla posizione dell'origine nel sistema di coordinate. Si noti che questo algoritmo è invariante alle trasformazioni ortogonali e alle traslazioni del sistema di coordinate; quindi la traslazione o le riflessioni del sistema di coordinate comportano la selezione delle stesse caselle da parte dell'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
. Ad esempio: selezionati_indices = tf.image.non_max_suppression(boxes, scores, max_output_size, iou_threshold) selezionati_boxes = tf.gather(boxes, selezionati_indices)
Argomenti:
- scope: un oggetto Scope
- boxs: un tensore float 2-D di forma
[num_boxes, 4]
. - punteggi: un tensore float 1-D di forma
[num_boxes]
che rappresenta un singolo punteggio corrispondente a ciascuna casella (ogni riga di caselle). - max_output_size: un tensore scalare intero che rappresenta il numero massimo di caselle da selezionare mediante la soppressione non massima.
Attributi facoltativi (vedi Attrs
):
- iou_threshold: Un float che rappresenta la soglia per decidere se le caselle si sovrappongono troppo rispetto all'IOU.
Resi:
-
Output
: un tensore intero 1-D di forma[M]
che rappresenta gli indici selezionati dal tensore delle caselle, doveM <= 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 pubbliche statiche | |
---|---|
IouThreshold (float x) |
Strutture | |
---|---|
tensorflow:: ops:: NonMaxSuppression:: Attrs | Setter di attributi facoltativi per NonMaxSuppression . |
Attributi pubblici
operazione
Operation operation
indici_selezionati
::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
operatore::tensorflow::Input
operator::tensorflow::Input() const
operatore::tensorflow::Output
operator::tensorflow::Output() const
Funzioni pubbliche statiche
IoSoglia
Attrs IouThreshold( float x )