Seleziona avidamente un sottoinsieme di riquadri di delimitazione in ordine decrescente di punteggio,
Questa operazione esegue non_max_suppression sugli input per batch, in tutte le classi. Elimina le caselle che presentano un'elevata sovrapposizione di intersezioni su unione (IOU) con le caselle selezionate in precedenza. I riquadri di delimitazione vengono forniti come [y1, x1, y2, x2], dove (y1, x1) e (y2, x2) sono le coordinate di qualsiasi coppia diagonale di angoli del riquadro e le coordinate possono essere fornite come normalizzate (ovvero, giacenti in l'intervallo [0, 1]) o assoluto. Si noti che questo algoritmo è indipendente dalla posizione dell'origine nel sistema di coordinate. Si noti inoltre che questo algoritmo è invariante alle trasformazioni ortogonali e alle traslazioni del sistema di coordinate; quindi la traslazione o le riflessioni del sistema di coordinate comportano la selezione delle stesse caselle da parte dell'algoritmo. L'output di questa operazione sono le caselle finali, i punteggi e il tensore delle classi restituiti dopo aver eseguito non_max_suppression.
Classi nidificate
| classe | CombinedNonMaxSuppression.Options | Attributi facoltativi per CombinedNonMaxSuppression | |
Metodi pubblici
| CombinedNonMaxSuppression.Options statico | clipBox (clipBox booleani) |
| Statico CombinedNonMaxSuppression | create ( ambito ambito , caselle operando <float>, punteggi operando <float>, operando <intero> maxOutputSizePerClass, operando <intero> maxTotalSize, operando <float> iouThreshold, operando <float> scoreThreshold, opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione CombinedNonMaxSuppression. |
| Uscita <Mobile> | nmsedBox () Un tensore [batch_size, max_detections, 4] float32 contenente le caselle soppresse non massime. |
| Uscita <Mobile> | nmsedClasses () Un tensore [batch_size, max_detections] float32 contenente le classi per le scatole. |
| Uscita <Mobile> | nmsedScores () Un tensore [batch_size, max_detections] float32 contenente i punteggi per le caselle. |
| CombinedNonMaxSuppression.Options statico | padPerClass (padPerClass booleano) |
| Output <Intero> | validDetections () Un tensore int32 [batch_size] che indica il numero di rilevamenti validi per articolo batch. |
Metodi ereditati
Metodi pubblici
public static CombinedNonMaxSuppression.Options clipBoxes (clipBox booleani)
Parametri
| clipBoxes | Se vero, presupponi che le coordinate della casella siano comprese tra [0, 1] e ritaglia le caselle di output se cadono oltre [0, 1]. Se falso, non eseguire il ritaglio e visualizzare le coordinate del riquadro così come sono. |
|---|
creazione statica pubblica CombinedNonMaxSuppression (ambito ambito , caselle operando <Float>, punteggi operando <Float>, operando <Integer> maxOutputSizePerClass, operando <Integer> maxTotalSize, operando <Float> iouThreshold, operando <Float> scoreThreshold, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione CombinedNonMaxSuppression.
Parametri
| scopo | ambito attuale |
|---|---|
| scatole | Un tensore float 4-D di forma "[batch_size, num_boxes, q, 4]". Se "q" è 1, vengono utilizzate le stesse caselle per tutte le classi, altrimenti, se "q" è uguale al numero di classi, vengono utilizzate caselle specifiche della 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). |
| maxOutputSizePerClass | Un tensore intero scalare che rappresenta il numero massimo di caselle da selezionare mediante soppressione non massima per classe |
| dimensione totale max | Uno scalare che rappresenta il numero massimo di caselle conservate in tutte le classi. |
| iouSoglia | Un tensore float 0-D che rappresenta la soglia per decidere se le scatole si sovrappongono troppo rispetto all'IOU. |
| punteggioSoglia | Un tensore float 0-D che rappresenta la soglia per decidere quando rimuovere le caselle in base al punteggio. |
| opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di CombinedNonMaxSuppression
Uscita pubblica <Float> nmsedBoxes ()
Un tensore [batch_size, max_detections, 4] float32 contenente le caselle soppresse non massime.
Output pubblico <Float> nmsedClasses ()
Un tensore [batch_size, max_detections] float32 contenente le classi per le scatole.
Output pubblico <Float> nmsedScores ()
Un tensore [batch_size, max_detections] float32 contenente i punteggi per le caselle.
public static CombinedNonMaxSuppression.Options padPerClass (padPerClass booleano)
Parametri
| padPerClass | Se false, le caselle, i punteggi e le classi di output nmsed vengono riempiti/ritagliati in "max_total_size". Se vero, le caselle, i punteggi e le classi di output nmsed vengono riempiti in modo da avere lunghezza `max_size_per_class`*`num_classes`, a meno che non superi `max_total_size`, nel qual caso viene ritagliato a `max_total_size`. Il valore predefinito è falso. |
|---|
output pubblico <numero intero> validDetections ()
Un tensore int32 [batch_size] che indica il numero di rilevamenti validi per articolo batch. Sono valide solo le prime voci num_detections[i] in nms_boxes[i], nms_scores[i] e nms_class[i]. Il resto delle voci sono imbottiture pari a zero.