LabelUtil

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
Arroja
IllegalArgumentException si tensorBuffer o labels es nulo.