public クラス LabelUtil
ラベル オペレーション ユーティリティ。
パブリック メソッド
static List<String> |
mapValueToLabels(TensorBuffer tensorBuffer, List<String> labels, int offset)
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 | インデックス値を持つテンソル。値は 0 以上の整数にする必要があります。各値 x は labels[x + offset] に変換されます。テンソルを浮動小数点数 TensorBuffer として指定すると、値は整数にキャストされます。範囲外の値はすべて空の文字列にマッピングされます。 |
---|---|
labels | 検索する辞書として使用する文字列のリスト。配列要素のインデックスがキーとして使用されます。パフォーマンスを向上させるには、ArrayList など、RandomAccess を実装するオブジェクトを使用します。 |
offset | labels で int 値を検索する際のオフセット値。 |
戻り値
- マッピングされた文字列。リストの長さは
TensorBuffer.getFlatSize()
です。
例外
IllegalArgumentException | tensorBuffer または labels が null の場合。 |
---|