tensoreflusso:: ops:: Soppressione combinata NonMax
#include <image_ops.h>
Seleziona avidamente un sottoinsieme di riquadri di delimitazione in ordine decrescente di punteggio.
Riepilogo
Questa operazione esegue non_max_suppression sugli input per batch, in tutte le classi. Elimina le caselle che presentano un'elevata sovrapposizione di intersezioni su unione (IOU) con le caselle selezionate in precedenza. 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 inoltre 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 sono le caselle finali, i punteggi e il tensore delle classi restituiti dopo aver eseguito non_max_suppression.
Argomenti:
- scope: un oggetto Scope
- boxs: un tensore float 4-D di forma
[batch_size, num_boxes, q, 4]
. Seq
è 1, vengono utilizzate le stesse caselle per tutte le classi, altrimenti, seq
è uguale al numero di classi, vengono utilizzate caselle specifiche della classe. - punteggi: un tensore float 3D di forma
[batch_size, num_boxes, num_classes]
che rappresenta un singolo punteggio corrispondente a ciascuna casella (ogni riga di caselle). - max_output_size_per_class: un tensore intero scalare che rappresenta il numero massimo di caselle da selezionare mediante soppressione non massima per classe
- max_total_size: uno scalare che rappresenta il numero massimo di caselle conservate in tutte le classi.
- iou_threshold: un tensore float 0-D che rappresenta la soglia per decidere se le scatole si sovrappongono troppo rispetto a IOU.
- score_threshold: un tensore float 0-D che rappresenta la soglia per decidere quando rimuovere le caselle in base al punteggio.
Attributi facoltativi (vedi Attrs
):
- pad_per_class: se false, le caselle, i punteggi e le classi di output nmsed vengono riempiti/ritagliati a
max_total_size
. Se vero, le caselle, i punteggi e le classi nms di output vengono riempiti per avere una lunghezzamax_size_per_class
*num_classes
, a meno che non superimax_total_size
nel qual caso viene ritagliato amax_total_size
. Il valore predefinito è falso. - clip_boxes: se vero, presuppone che le coordinate della casella siano comprese tra [0, 1] e ritaglia le caselle di output se cadono oltre [0, 1]. Se falso, non eseguire il ritaglio e visualizzare le coordinate del riquadro così come sono.
Resi:
-
Output
nmsed_boxes: un tensore [batch_size, max_detections, 4] float32 contenente i riquadri soppressi non massimi. -
Output
nmsed_scores: un tensore [batch_size, max_detections] float32 contenente i punteggi per le caselle. -
Output
nmsed_classes: un tensore [batch_size, max_detections] float32 contenente le classi per le scatole. -
Output
valid_detections: un tensore [batch_size] int32 che indica il numero di rilevamenti validi per articolo batch. Sono valide solo le prime voci num_detections[i] in nms_boxes[i], nms_scores[i] e nms_class[i]. Il resto delle voci sono imbottiture pari a zero.
Costruttori e distruttori | |
---|---|
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) |
Attributi pubblici | |
---|---|
nmsed_boxes | |
nmsed_classes | |
nmsed_scores | |
operation | |
valid_detections |
Funzioni pubbliche statiche | |
---|---|
ClipBoxes (bool x) | |
PadPerClass (bool x) |
Strutture | |
---|---|
tensorflow:: ops:: CombinedNonMaxSuppression:: Attrs | Setter di attributi facoltativi per CombinedNonMaxSuppression . |
Attributi pubblici
nmsed_boxes
::tensorflow::Output nmsed_boxes
nmsed_classes
::tensorflow::Output nmsed_classes
nmsed_scores
::tensorflow::Output nmsed_scores
operazione
Operation operation
rilevamenti_validi
::tensorflow::Output valid_detections
Funzioni pubbliche
Soppressione combinata NonMax
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 )
Soppressione combinata NonMax
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 )
Funzioni pubbliche statiche
ClipBox
Attrs ClipBoxes( bool x )
PadPerClass
Attrs PadPerClass( bool x )