tensorflow :: ops :: NonMaxSuppressionWithOverlaps

#include <image_ops.h>

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

Sommario

potatura di cassette che presentano alte sovrapposizioni con cassette precedentemente selezionate. I riquadri di delimitazione con un punteggio inferiore a score_threshold vengono rimossi. I valori di sovrapposizione N per n sono forniti come matrice quadrata, che consente di definire un criterio di sovrapposizione personalizzato (es. Intersezione su unione, intersezione su area, ecc.).

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 . Per esempio:

selected_indices = tf.image.non_max_suppression_with_overlaps (overlaps, score, max_output_size, overlap_threshold, score_threshold) selected_boxes = tf.gather (box, selected_indices)

Argomenti:

  • scope: un oggetto Scope
  • overlaps: un tensore [num_boxes, num_boxes] virgola mobile di forma [num_boxes, num_boxes] rappresenta i valori di sovrapposizione n-by-n box.
  • 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.
  • overlap_threshold: un tensore a virgola mobile 0-D che rappresenta la soglia per decidere se anche le caselle si sovrappongono.
  • score_threshold: un tensore a virgola mobile 0-D che rappresenta la soglia per decidere quando rimuovere i box in base al punteggio.

Ritorna:

  • Output : Un tensore intero 1-D di forma [M] rappresenta gli indici selezionati dal tensore delle caselle, dove M <= max_output_size .

Costruttori e distruttori

NonMaxSuppressionWithOverlaps (const :: tensorflow::Scope & scope, :: tensorflow::Input overlaps, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input overlap_threshold, :: tensorflow::Input score_threshold)

Attributi pubblici

operation
selected_indices

Funzioni pubbliche

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Attributi pubblici

operazione

Operation operation

selected_indices

::tensorflow::Output selected_indices

Funzioni pubbliche

NonMaxSuppressionWithOverlaps

 NonMaxSuppressionWithOverlaps(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input overlaps,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input overlap_threshold,
  ::tensorflow::Input score_threshold
)

nodo

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const