LabelUtil

публичный класс LabelUtil

Утилиты работы с метками.

Публичные методы

статический список < строка >
mapValueToLabels ( TensorBuffer tensorBuffer, List <String> метки, смещение int)
Сопоставляет тензор значений int со списком строковых меток.

Унаследованные методы

Публичные методы

общедоступный статический список <String> mapValueToLabels ( TensorBuffer tensorBuffer, List < String > labels, int offset)

Сопоставляет тензор значений int со списком строковых меток. В качестве словаря он принимает массив строк. Пример: если задан тензор [3, 1, 0] и заданы метки ["фон", "яблоко", "банан", "вишня", "дата"], результатом будет ["дата", «банан», «яблоко»].

Параметры
ТензорБуфер Тензор со значениями индекса. Значения должны быть неотрицательными целыми числами, и каждое значение x будет преобразовано в labels[x + offset] . Если тензор задан как число с плавающей запятой TensorBuffer , значения будут преобразованы в целые числа. Все значения, выходящие за пределы, будут отображаться в пустую строку.
этикетки Список строк, используемый в качестве словаря для поиска. В качестве ключа будет использоваться индекс элемента массива. Чтобы повысить производительность, используйте объект, реализующий RandomAccess, например ArrayList .
компенсировать Значение смещения при поиске целочисленных значений в labels .
Возврат
Броски
IllegalArgumentException если tensorBuffer или labels имеют значение null.