classe pubblica LabelUtil
Utilità per l'operazione di etichetta.
Metodi pubblici
Elenco statico < String > | mapValueToLabels ( TensorBuffer tensorBuffer, List < String > etichette, int offset) Mappa un tensore di valori int su un elenco di etichette di stringhe. |
Metodi ereditati
Metodi pubblici
public static List < String > mapValueToLabels ( TensorBuffer tensorBuffer, List < String > label, int offset)
Mappa un tensore di valori int su un elenco di etichette di stringhe. Richiede un array di stringhe come dizionario. Esempio: se il tensore dato è [3, 1, 0] e le etichette date sono ["sfondo", "mela", "banana", "ciliegia", "data"], il risultato sarà ["data", "banana", "mela"].
Parametri
tensorBuffer | Un tensore con valori di indice. I valori devono essere numeri interi non negativi e ogni valore x verrà convertito in labels[x + offset] . Se il tensore viene fornito come float TensorBuffer , i valori verranno convertiti in numeri interi. Tutti i valori fuori limite verranno mappati su una stringa vuota. |
---|---|
etichette | Un elenco di stringhe, utilizzato come dizionario da cercare. L'indice dell'elemento dell'array verrà utilizzato come chiave. Per ottenere prestazioni migliori, utilizzare un oggetto che implementa RandomAccess, come ArrayList . |
compensare | Il valore di offset quando si cercano valori int nelle labels . |
ritorna
- le stringhe mappate. La lunghezza dell'elenco è
TensorBuffer.getFlatSize()
.
Lancia
IllegalArgumentException | se tensorBuffer o labels sono nulli. |
---|