classe pública LabelUtil
Utilitários da operação de identificador.
Métodos públicos
estático Lista<String> |
mapValueToLabels(TensorBuffer, tensorBuffer, List<String> rótulos, int offset)
Mapeia um tensor de valor int para uma lista de rótulos de 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
- as strings mapeadas. O tamanho da lista é
TensorBuffer.getFlatSize()
.
Gera
IllegalArgumentException | se tensorBuffer ou labels for nulo.
|
---|