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 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.