tensorflow::ops::NonMaxSuppressionV2

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

Riepilogo

Eliminazione dei riquadri che presentano un'elevata sovrapposizione di intersezione-su-unione (IOU) con i riquadri precedentemente selezionati. I riquadri di delimitazione sono forniti come [y1, x1, y2, x2], dove (y1, x1) e (y2, x2) sono le coordinate di una qualsiasi coppia diagonale di angoli del riquadro e le coordinate possono essere fornite normalizzate (ovvero, comprese nell'intervallo [0, 1]) o assolute. Si noti che questo algoritmo è indipendente dalla posizione dell'origine nel sistema di coordinate. Si noti inoltre che questo algoritmo è invariante rispetto a trasformazioni ortogonali e traslazioni del sistema di coordinate; pertanto, traslazioni o riflessioni del sistema di coordinate portano alla selezione degli stessi riquadri da parte dell'algoritmo.

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_v2( boxs, scores, max_output_size, iou_threshold) selected_boxes = tf.gather(boxes, selected_indices)

Argomentazioni:

• ambito: un oggetto Scope
• scatole: un tensore float 2D di forma [num_boxes, 4] .
• 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.
• iou_threshold: Un tensore float 0-D che rappresenta la soglia per decidere se le caselle si sovrappongono troppo rispetto all'IOU.

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

 NonMaxSuppressionV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const