Classe publique LabelUtil
Utilitaires d'opération d'étiquetage
Méthodes publiques
statique List<String> |
mapValueToLabels(TensorBuffer tensorBuffer, List<String> étiquettes, décalage entier)
Mappe un Tensor de valeur int à une liste d'étiquettes de chaîne.
|
Méthodes héritées
Méthodes publiques
public statique List<String> mapValueToLabels (TensorBuffer tensorBuffer, List<String> étiquettes, décalage entier)
Mappe un Tensor de valeur int à une liste d'étiquettes de chaîne. Le dictionnaire utilise un tableau de chaînes. Exemple: si le Tensor donné est [3, 1, 0] et que les étiquettes données sont ["arrière-plan", "pomme", "banane", "cerise", "date"], le résultat sera ["date", "banane", "pomme"].
Paramètres
tensorBuffer | Tensor avec des valeurs d'index. Les valeurs doivent être des entiers non négatifs, et chaque valeur x sera convertie en labels[x + offset] . Si le Tensor est donné en tant que TensorBuffer flottant, les valeurs seront converties en entiers. Toutes les valeurs hors limites seront mappées à une chaîne vide. |
---|---|
étiquettes | Liste de chaînes, utilisée comme dictionnaire à rechercher. L'index de l'élément du tableau sera utilisé comme clé. Pour de meilleures performances, utilisez un objet qui implémente RandomAccess, tel que ArrayList . |
offset | Valeur de décalage lors de la recherche de valeurs entières dans labels . |
Renvoie
- les chaînes mappées. La longueur de la liste est de
TensorBuffer.getFlatSize()
.
Génère
IllegalArgumentException | si tensorBuffer ou labels est nul.
|
---|