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] . Se q è 1, vengono utilizzate le stesse caselle per tutte le classi, altrimenti, se q è 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 lunghezza max_size_per_class * num_classes , a meno che non superi max_total_size nel qual caso viene ritagliato a max_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.

Ritorna:

  • 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
)