LabelUtil

公共类 LabelUtil

标签操作实用程序

公共方法

static List<String>
mapValueToLabels(TensorBuffer tensorBuffer, List<String> 标签, int 偏移)
将 int 值张量映射到字符串标签列表。

继承的方法

公共方法

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

将 int 值张量映射到字符串标签列表。它接受一个字符串数组作为字典。示例：如果给定张量为 [3, 1, 0]，并且给定标签为 ["background", "apple", "banana", "cherry", "date"]，结果将为 ["date", "banana", "apple"]。

参数
tensorBuffer 具有索引值的张量。值应为非负整数，且每个值 x 都将转换为 labels[x + offset]。如果以浮点 TensorBuffer 的形式提供张量，则值将转换为整数。所有超出范围的值都将映射到空字符串。
标签 一个字符串列表，可用作要查找的字典。数组元素的索引将用作键。为了获得更好的性能，请使用实现 RandomAccess 的对象，例如 ArrayList。
offset 在 labels 中查找 int 值时的偏移值。
返回
抛出
IllegalArgumentException 如果 tensorBuffer 或 labels 为 null。