tensorflow::ops::NonMaxSuppressionWithOverlaps

Seleziona avidamente un sottoinsieme di bounding box in ordine decrescente di punteggio.

Riepilogo

Vengono eliminate le caselle che presentano un'elevata sovrapposizione con le caselle precedentemente selezionate. Vengono rimosse anche le caselle di delimitazione con punteggio inferiore a score_threshold . I valori di sovrapposizione N x N vengono forniti come matrice quadrata, che consente di definire un criterio di sovrapposizione personalizzato (ad esempio, intersezione sull'unione, intersezione sull'area, ecc.).

L'output di questa operazione è un insieme di numeri interi che indicizzano la collezione di input di bounding box che rappresentano i riquadri selezionati. Le coordinate delle bounding box corrispondenti agli indici selezionati possono quindi essere ottenute utilizzando l' tf.gather operation . Ad esempio:

selected_indices = tf.image.non_max_suppression_with_overlaps( overlaps, scores, max_output_size, overlap_threshold, score_threshold) selected_boxes = tf.gather(boxes, selected_indices)

Argomentazioni:

• ambito: un oggetto Scope
• sovrapposizioni: un tensore float 2D di forma [num_boxes, num_boxes] che rappresenta i valori di sovrapposizione delle caselle n x n.
• punteggi: Un tensore float 1D di forma [num_boxes] che rappresenta un singolo punteggio corrispondente a ciascuna casella (ogni riga di caselle).
• max_output_size: Un tensore scalare intero che rappresenta il numero massimo di riquadri da selezionare tramite soppressione non massima.
• overlap_threshold: Un tensore float 0-D che rappresenta la soglia per decidere se anche le caselle si sovrappongono.
• score_threshold: Un tensore float 0-D che rappresenta la soglia per decidere quando rimuovere le caselle in base al punteggio.

Resi:

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

attributi pubblici

Operation operation

::tensorflow::Output selected_indices

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

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const