tensorflow::ops::CombinedNonMaxSuppression

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

Riepilogo

Questa operazione esegue la soppressione non-max sugli input per batch, su tutte le classi. Elimina i riquadri che presentano un'elevata sovrapposizione di intersezione-su-unione (IOU) con i riquadri precedentemente selezionati. I riquadri di delimitazione vengono 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 producono la selezione degli stessi riquadri da parte dell'algoritmo. L'output di questa operazione è il tensore finale di riquadri, punteggi e classi restituito dopo l'esecuzione della soppressione non-max.

Argomentazioni:

• ambito: un oggetto Scope
• box: un tensore float 4D di forma [batch_size, num_boxes, q, 4] . Se q è 1, vengono utilizzate le stesse box per tutte le classi; altrimenti, se q è uguale al numero di classi, vengono utilizzate box specifiche per ciascuna classe.
• punteggi: Un tensore float 3D di forma [batch_size, num_boxes, num_classes] che rappresenta un singolo punteggio corrispondente a ciascuna casella (ogni riga di caselle).
• max_output_size_per_class: Un tensore intero scalare che rappresenta il numero massimo di caselle da selezionare tramite soppressione non massima per classe
• max_total_size: Un valore scalare che rappresenta il numero massimo di box mantenute per tutte le classi.
• iou_threshold: Un tensore float 0-D che rappresenta la soglia per decidere se le caselle si sovrappongono troppo rispetto all'IOU.
• score_threshold: Un tensore float 0-D che rappresenta la soglia per decidere quando rimuovere le caselle in base al punteggio.

Attributi opzionali (vedere Attrs ):

• pad_per_class: Se impostato su false, i riquadri, i punteggi e le classi nmsed di output vengono riempiti/troncati a max_total_size . Se impostato su true, i riquadri, i punteggi e le classi nmsed di output vengono riempiti fino a raggiungere una lunghezza pari max_size_per_class * num_classes , a meno che non superino max_total_size , nel qual caso vengono troncati a max_total_size . Il valore predefinito è false.
• clip_boxes: Se true, si presume che le coordinate del riquadro siano comprese tra [0, 1] e si ritagliano i riquadri di output se si trovano al di fuori di [0, 1]. Se false, non si esegue il ritaglio e si restituiscono le coordinate del riquadro così come sono.

Resi:

• Output nmsed_boxes: un tensore float32 di tipo [batch_size, max_detections, 4] contenente le box soppresse non massime.
• Output nmsed_scores: un tensore float32 di dimensioni [batch_size, max_detections] contenente i punteggi per le caselle.
• Output nmsed_classes: un tensore float32 di dimensioni [batch_size, max_detections] contenente le classi per le caselle.
• Output valid_detections: un tensore int32 di dimensione [batch_size] che indica il numero di rilevamenti validi per elemento del batch. Solo le prime num_detections[i] voci in nms_boxes[i], nms_scores[i] e nms_class[i] sono valide. Le restanti voci sono riempite con zeri.

attributi pubblici

::tensorflow::Output nmsed_boxes

::tensorflow::Output nmsed_classes

::tensorflow::Output nmsed_scores

Operation operation

::tensorflow::Output valid_detections

 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
)

Attrs ClipBoxes(
  bool x
)

Attrs PadPerClass(
  bool x
)