Model

パブリッククラスモデル

TFLite モデルおよび TFLite インタープリターのラッパー クラス。

注: Model一度に 1 つの TFLite モデルのみを保持でき、それを実行するための TFLite インタープリター インスタンスを常に保持します。

ネストされたクラス

クラスモデル.ビルダーこのクラスは廃止されました。 Model.createModel(Context, String, Options)を使用してください。
列挙型モデル.デバイス分類の実行に使用されるランタイムデバイスのタイプ。
クラスモデル.オプションモデルを実行するためのオプション。

パブリックメソッド

空所
近い()
静的モデル
createModel (Context コンテキスト、 String modelPath、 Model.Optionsオプション)
アセットからモデルをロードし、指定されたオプションを使用して TFLite インタープリターを初期化します。
静的モデル
createModel (コンテキスト context、文字列modelPath)
アセットからモデルをロードし、TFLite インタープリターを初期化します。
マップされたバイトバッファ
getData ()
メモリマップされたモデルデータを返します。
テンソル
getInputTensor (int inputIndex)
指定された入力インデックスに関連付けられた Tensor を取得します。
テンソル
getOutputTensor (int OutputIndex)
指定された出力インデックスに関連付けられた Tensor を取得します。
int[]
getOutputTensorShape (int OutputIndex)
出力形状を返します。
getPath ()
Assets に保存されているモデル ファイルのパスを返します。
空所
run ( Object[]入力、 Map < IntegerObject > 出力)
複数の入力に対してモデル推論を実行し、複数の出力を返します。

継承されたメソッド

パブリックメソッド

public void close ()

public static Model createModel (コンテキストコンテキスト、文字列モデルパス、 Model.Optionsオプション)

アセットからモデルをロードし、指定されたオプションを使用して TFLite インタープリターを初期化します。

パラメーター
コンテクストアプリのコンテキスト。
モデルパスモデルファイルのパス。
オプションモデルを実行するためのオプション。
投げる
IO例外モデル ファイルを開くときに例外が発生した場合。
関連項目

public static Model createModel (コンテキスト context, String modelPath)

アセットからモデルをロードし、TFLite インタープリターを初期化します。

デフォルトのオプションは次のとおりです: (1) CPU デバイス。 (2) 1 つのスレッド。

パラメーター
コンテクストアプリのコンテキスト。
モデルパスモデルファイルのパス。
投げる
IO例外モデル ファイルを開くときに例外が発生した場合。

public MappedByteBuffer getData ()

メモリマップされたモデルデータを返します。

public Tensor getInputTensor (int inputIndex)

指定された入力インデックスに関連付けられた Tensor を取得します。

パラメーター
入力インデックス
投げる
IllegalStateExceptionインタプリタが閉じている場合。

public Tensor getOutputTensor (int OutputIndex)

指定された出力インデックスに関連付けられた Tensor を取得します。

パラメーター
出力インデックス
投げる
IllegalStateExceptionインタプリタが閉じている場合。

public int[] getOutputTensorShape (int OutputIndex)

出力形状を返します。出力形状がグラフの作成時にのみ決定される場合に便利です。

パラメーター
出力インデックス
投げる
IllegalStateExceptionインタプリタが閉じている場合。

パブリックString getPath ()

Assets に保存されているモデル ファイルのパスを返します。

public void run ( Object[]入力、 Map < IntegerObject > 出力)

複数の入力に対してモデル推論を実行し、複数の出力を返します。

パラメーター
入力入力データの配列。入力はモデルの入力と同じ順序である必要があります。各入力は、配列または多次元配列、または int、float、long、byte などのプリミティブ型のByteBufferにすることができます。 ByteBufferは大きな入力データを渡すための推奨される方法ですが、文字列型では (多次元) 配列入力パスを使用する必要があります。 ByteBufferが使用される場合、モデル推論が完了するまでその内容は変更されないはずです。
出力出力インデックスを出力データの多次元配列、または int、float、long、byte などのプリミティブ型のByteBufferにマッピングするマップ。使用する出力のエントリを保持するだけで済みます。