tensorflow :: ops :: NonMaxSuppressionV5

#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 selezionate in precedenza. I riquadri di delimitazione con un punteggio inferiore a score_threshold vengono rimossi. 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. Si noti che questo algoritmo è agnostico rispetto a dove l'origine è nel sistema di coordinate e più generalmente è 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 . Ad esempio: selected_indices = tf.image.non_max_suppression_v2 (box, score, max_output_size, iou_threshold, score_threshold) selected_boxes = tf.gather (box, selected_indices) Questa opzione supporta anche una modalità Soft-NMS (con ponderazione gaussiana) (cfr Bodla et al , https://arxiv.org/abs/1704.04503 ) dove le scatole riducono il punteggio di altre scatole sovrapposte invece di provocarne l'eliminazione diretta. Per abilitare questa modalità Soft-NMS, impostare il parametro soft_nms_sigma su un valore maggiore di 0.

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.
  • score_threshold: un tensore a virgola mobile 0-D che rappresenta la soglia per decidere quando rimuovere le caselle in base al punteggio.
  • soft_nms_sigma: un tensore a virgola mobile 0-D che rappresenta il parametro sigma per Soft NMS; vedere Bodla et al (cfr https://arxiv.org/abs/1704.04503 ). Quando soft_nms_sigma=0.0 (che è l'impostazione predefinita), torniamo allo standard (hard) NMS.

Attributi opzionali (vedi Attrs ):

  • pad_to_max_output_size: se true, l'output selected_indices viene riempito in modo da essere di lunghezza max_output_size . Il valore predefinito è false.

Ritorna:

  • Output selected_indices: Un tensore intero 1-D di forma [M] rappresenta gli indici selezionati dalle caselle tensore, dove M <= max_output_size .
  • Output selected_scores: un tensore a virgola mobile 1-D di forma [M] rappresenta i punteggi corrispondenti per ciascuna casella selezionata, dove M <= max_output_size . I punteggi differiscono solo dai punteggi di input corrispondenti quando si utilizza Soft NMS (ovvero quando soft_nms_sigma>0 )
  • Output valid_outputs: un tensore intero 0-D che rappresenta il numero di elementi validi in selected_indices , con gli elementi validi che compaiono per primi.

Costruttori e distruttori

NonMaxSuppressionV5 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, :: tensorflow::Input soft_nms_sigma)
NonMaxSuppressionV5 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, :: tensorflow::Input soft_nms_sigma, const NonMaxSuppressionV5::Attrs & attrs)

Attributi pubblici

operation
selected_indices
selected_scores
valid_outputs

Funzioni statiche pubbliche

PadToMaxOutputSize (bool x)

Structs

tensorflow :: ops :: NonMaxSuppressionV5 :: Attrs

Setter di attributi facoltativi per NonMaxSuppressionV5 .

Attributi pubblici

operazione

Operation operation

selected_indices

::tensorflow::Output selected_indices

selected_scores

::tensorflow::Output selected_scores

valid_outputs

::tensorflow::Output valid_outputs

Funzioni pubbliche

NonMaxSuppressionV5

 NonMaxSuppressionV5(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  ::tensorflow::Input soft_nms_sigma
)

NonMaxSuppressionV5

 NonMaxSuppressionV5(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  ::tensorflow::Input soft_nms_sigma,
  const NonMaxSuppressionV5::Attrs & attrs
)

Funzioni statiche pubbliche

PadToMaxOutputSize

Attrs PadToMaxOutputSize(
  bool x
)