clase pública LabelUtil
Utilidades de operación de etiquetas
Métodos públicos
estático List<String> |
mapValueToLabels(TensorBuffer tensorBuffer, List<String> etiquetas, int offset)
Asigna un tensor de valor int a una lista de etiquetas de string.
|
Métodos heredados
Métodos públicos
publica estática estática List<String> mapValueToLabels (TensorBuffer tensorBuffer, List<String> labels, int offset)
Asigna un tensor de valor int a una lista de etiquetas de string. Toma un array de cadenas como diccionario. Ejemplo: Si el tensor dado es [3, 1, 0] y las etiquetas dadas es ["background", "apple", "banana", "cherry", "date"], el resultado será ["date", "banana", "apple"].
Parámetros
tensorBuffer | Un tensor con valores de índice. Los valores deben ser números enteros no negativos, y cada valor x se convertirá en labels[x + offset] . Si el tensor se proporciona como un TensorBuffer flotante, los valores se convertirán en números enteros. Todos los valores que estén fuera de límite se asignarán a una string vacía. |
---|---|
etiquetas | Una lista de cadenas, que se usa como diccionario para buscar. Se usará el índice del elemento
del array como clave. Para obtener un mejor rendimiento, usa un objeto que implemente RandomAccess, como ArrayList . |
offset | El valor de desplazamiento cuando se buscan valores int en labels . |
Devuelve
- las cadenas asignadas. La longitud de la lista es de
TensorBuffer.getFlatSize()
.
Arroja
IllegalArgumentException | si tensorBuffer o labels es nulo.
|
---|