LabelUtil

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
Lancia
IllegalArgumentException se tensorBuffer o labels sono nulli.