tensorflow :: ops :: CombinedNonMaxSuppression

#include <image_ops.h>

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

Sommario

Questa operazione esegue non_max_suppression sugli input per batch, in tutte le classi. Elimina le scatole che hanno un'elevata sovrapposizione di intersezione su unione (IOU) con le 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 è nel sistema di coordinate. Si noti inoltre 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 sono le caselle finali, i punteggi e il tensore delle classi restituiti dopo aver eseguito non_max_suppression.

Argomenti:

  • scope: un oggetto Scope
  • box: 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 a virgola mobile 3-D di forma [batch_size, num_boxes, num_classes] rappresenta un singolo punteggio corrispondente a ciascuna casella (ciascuna 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 a virgola mobile 0-D che rappresenta la soglia per decidere se le caselle si sovrappongono troppo rispetto a IOU.
  • score_threshold: un tensore a virgola mobile 0-D che rappresenta la soglia per decidere quando rimuovere le caselle in base al punteggio.

Attributi opzionali (vedi Attrs ):

  • pad_per_class: se false, le caselle, i punteggi e le classi nmsed di output vengono riempiti / ritagliati a max_total_size . Se TRUE, l'uscita nmsed scatole, colonne e le classi sono imbottiti essere di lunghezza max_size_per_class * num_classes , a meno che non supera max_total_size nel qual caso viene troncato max_total_size . Il valore predefinito è false.
  • clip_boxes: se true, supponi che le coordinate del riquadro siano comprese tra [0, 1] e ritaglia le caselle di output se cadono oltre [0, 1]. Se false, non ritagliare e visualizzare le coordinate della casella così com'è.

Ritorna:

  • Output nmsed_boxes: Un tensore [batch_size, max_detections, 4] float32 contenente le caselle non-max soppresse.
  • Output nmsed_scores: un tensore float32 [batch_size, max_detections] contenente i punteggi per le caselle.
  • Output nmsed_classes: un tensore [batch_size, max_detections] float32 contenente le classi per le caselle.
  • Output valid_detections: un tensore [batch_size] int32 che indica il numero di rilevamenti validi per elemento batch. Sono valide solo le prime num_detections [i] voci in nms_boxes [i], nms_scores [i] e nms_class [i]. Il resto delle voci sono zero padding.

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 statiche pubbliche

ClipBoxes (bool x)
PadPerClass (bool x)

Structs

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

valid_detections

::tensorflow::Output valid_detections

Funzioni pubbliche

CombinedNonMaxSuppression

 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

 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 statiche pubbliche

ClipBoxes

Attrs ClipBoxes(
  bool x
)

PadPerClass

Attrs PadPerClass(
  bool x
)