GenerateBoundingBoxProposals

classe finale pubblica GenerateBoundingBoxProposals

Questa operazione produce la regione di interesse da determinati riquadri di delimitazione (bbox_deltas) codificati rispetto agli ancoraggi secondo l'eq.2 in arXiv:1506.01497

L'operazione seleziona le prime caselle di punteggio "pre_nms_topn", le decodifica rispetto alle ancore, applica la soppressione non massimale su caselle sovrapposte con un valore di intersezione su unione (iou) superiore a "nms_threshold", scartando le caselle in cui il lato più corto è inferiore a " dimensione_min`. Input: `scores`: un tensore di forma 4D [Batch, Altezza, Larghezza, Num Ancore] contenente i punteggi per ancora in una determinata posizione `bbox_deltas`: è un tensore di forma [Batch, Altezza, Larghezza, 4 x Num Ancore] riquadri codificati per ciascuna ancora "ancore": un tensore 1D di forma [4 x Num ancore], che rappresenta le ancore. Output: `rois`: output RoIs, un tensore 3D di forma [Batch, post_nms_topn, 4], riempito con 0 se trovati meno dei candidati post_nms_topn. `roi_probabilities`: punteggi di probabilità di ciascun roi in 'rois', un tensore 2D di forma [Batch,post_nms_topn], riempito con 0 se necessario, ordinato per punteggi.

Classi nidificate

classe GeneraBoundingBoxProposals.Options Attributi facoltativi per GenerateBoundingBoxProposals

Metodi pubblici

proposte statiche GenerateBoundingBox
create ( Ambito di applicazione , punteggi dell'operando <Float>, operando <Float> bboxDeltas, operando <Float> imageInfo, ancoraggi dell'operando <Float>, operando <Float> nmsThreshold, operando <Integer> preNmsTopn, operando <Float> minSize , Opzioni.. . opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione GenerateBoundingBoxProposals.
static GenerateBoundingBoxProposals.Options
postNmsTopn (PostNmsTopn lungo)
Uscita <Mobile>
roiProbabilità ()
Un tensore float 2D di forma "[num_images, post_nms_topn]" che rappresenta il punteggio del riquadro della regione di interesse nel tensore "rois" con lo stesso indice.
Uscita <Mobile>
Rois ()
Un tensore float 3D di forma "[num_images,post_nms_topn,4]" che rappresenta la regione selezionata dei riquadri di interesse.

Metodi ereditati

Metodi pubblici

public static GenerateBoundingBoxProposals create ( Scope scope, Operando <Float> punteggi, Operando <Float> bboxDeltas, Operando <Float> imageInfo, Operando <Float> ancoraggi, Operando <Float> nmsThreshold, Operando <Integer> preNmsTopn, Operando <Float> minSize, Opzioni... opzioni)

Metodo factory per creare una classe che racchiude una nuova operazione GenerateBoundingBoxProposals.

Parametri
scopo ambito attuale
punteggi Un tensore float 4-D di forma `[num_images, altezza, larghezza, num_achors]` contenente i punteggi dei riquadri per determinate ancore, può essere non ordinato.
bboxDeltas Un tensore float 4-D di forma "[num_images, altezza, larghezza, 4 x num_anchors]". scatole di codifica con rispetto a ciascun ancoraggio. Le coordinate sono fornite nella forma [dy, dx, dh, dw].
immagineInfo Un tensore float 2D di forma "[num_images, 5]" contenente informazioni sull'immagine: altezza, larghezza e scala.
ancore Un tensore float 2-D di forma "[num_anchors, 4]" che descrive i riquadri di ancoraggio. Le caselle sono formattate nel formato [y1, x1, y2, x2].
nmsSoglia Un tensore scalare float per una soglia di soppressione non massimale.
preNmsTopn Un tensore scalare intero per il numero di caselle con punteggio massimo da utilizzare come input.
dimensione min Un tensore scalare float. Qualsiasi casella di dimensione inferiore a min_size verrà scartata.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di GenerateBoundingBoxProposals

public static GenerateBoundingBoxProposals.Options postNmsTopn (postNmsTopn lungo)

Parametri
postNmsTopn Un numero intero. Numero massimo di rois in uscita.

output pubblico <Float> roiProbabilities ()

Un tensore float 2D di forma "[num_images, post_nms_topn]" che rappresenta il punteggio del riquadro della regione di interesse nel tensore "rois" con lo stesso indice.

output pubblico <Float> rois ()

Un tensore float 3D di forma "[num_images,post_nms_topn,4]" che rappresenta la regione selezionata dei riquadri di interesse. Ordinati in ordine decrescente in punteggi.