classe publique LabelUtil
Utilitaires d’opération d’étiquette.
Méthodes publiques
Liste statique < Chaîne > | mapValueToLabels ( TensorBuffer tensorBuffer, List < String > labels, int offset) Mappe un tenseur de valeur int à une liste d’étiquettes de chaîne. |
Méthodes héritées
Méthodes publiques
public static List < String > mapValueToLabels ( TensorBuffer tensorBuffer, List < String > labels, int offset)
Mappe un tenseur de valeur int à une liste d’étiquettes de chaîne. Il prend un tableau de chaînes comme dictionnaire. Exemple : si le tenseur donné est [3, 1, 0] et que les étiquettes données sont ["background", "apple", "banana", "cherry", "date"], le résultat sera ["date", "banane", "pomme"].
Paramètres
tensorBuffer | Un tenseur 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 tenseur est donné sous forme de float TensorBuffer , les valeurs seront converties en entiers. Toutes les valeurs hors limites seront mappées à une chaîne vide. |
---|---|
Étiquettes | Une liste de chaînes, utilisée comme dictionnaire à rechercher. L'index de l'élément du tableau sera utilisé comme clé. Pour obtenir de meilleures performances, utilisez un objet qui implémente RandomAccess, tel que ArrayList . |
compenser | La valeur de décalage lors de la recherche des valeurs int dans les labels . |
Retour
- les chaînes mappées. La longueur de la liste est
TensorBuffer.getFlatSize()
.
Jetés
Exception d'argument illégal | si tensorBuffer ou labels est nul. |
---|