LabelUtil

classe pública LabelUtil

Utilitários da operação de identificador.

Métodos públicos

estático Lista<String>

Métodos herdados

Métodos públicos

public Static List<String> mapValueToLabels (TensorBuffer tensorBuffer, List<String> labels, int offset)

Mapeia um tensor de valor int para uma lista de rótulos de string. Ela usa uma matriz de strings como dicionário. Exemplo: se o tensor fornecido for [3, 1, 0], e os rótulos fornecidos forem ["background", "apple", "banana", "cherry", "date"], o resultado será ["date", "banana", "apple"].

Parâmetros
tensorBuffer Um tensor com valores de índice. Os valores precisam ser números inteiros não negativos, e cada valor x será convertido em labels[x + offset]. Se o tensor for fornecido como um TensorBuffer flutuante, os valores serão convertidos em números inteiros. Todos os valores que estiverem fora do limite serão mapeados para uma string vazia.
rótulos Uma lista de strings, usada como um dicionário para pesquisar. O índice do elemento da matriz será usado como a chave. Para melhorar o desempenho, use um objeto que implemente RandomAccess, como ArrayList.
offset O valor de deslocamento ao procurar valores int no labels.
Retorna
Gera
IllegalArgumentException se tensorBuffer ou labels for nulo.