パブリック クラス Model
TFLite モデルと TFLite インタープリタのラッパークラス。
注: Model
は一度に 1 つの TFLite モデルのみを保持でき、実行するための TFLite インタープリタ インスタンスを常に保持します。
ネストされたクラス
クラス | Model.Builder |
このクラスは非推奨になりました。Model.createModel(Context, String, Options) を使用してください。
|
|
enum | Model.Device | 分類の実行に使用されるランタイム デバイスタイプ。 | |
クラス | Model.Options | モデルを実行するためのオプション。 |
パブリック メソッド
void |
close()
|
static モデル |
createModel(コンテキスト コンテキスト、String modelPath、Model.Options オプション)
アセットからモデルを読み込み、指定されたオプションで TFLite インタープリタを初期化します。
|
static モデル | |
MappedByteBuffer |
getData()
メモリマップされたモデルデータを返します。
|
Tensor |
getInputTensor(int inputIndex)
指定された入力インデックスに関連付けられたテンソルを取得します。
|
Tensor |
getOutputTensor(int outputIndex)
指定された出力インデックスに関連付けられたテンソルを取得します。
|
int[] |
getOutputTensorShape(int outputIndex)
出力シェイプを返します。
|
文字列 |
getPath()
Assets に保存されているモデルファイルのパスを返します。
|
void |
継承されるメソッド
パブリック メソッド
public void close ()
public static Model createModel (コンテキスト コンテキスト、String modelPath、Model.Options オプション)
アセットからモデルを読み込み、指定されたオプションで TFLite インタープリタを初期化します。
パラメータ
コンテキスト | アプリのコンテキスト。 |
---|---|
modelPath | モデルファイルのパス。 |
オプション | モデルを実行するためのオプション。 |
例外
IOException | モデルファイルを開くときに例外が発生した場合。 |
---|
関連情報
public static モデル createModel (コンテキスト コンテキスト、String modelPath)
アセットからモデルを読み込み、TFLite インタープリタを初期化します。
デフォルトのオプションは、(1)CPU デバイス、(2)1 スレッドです。
パラメータ
コンテキスト | アプリのコンテキスト。 |
---|---|
modelPath | モデルファイルのパス。 |
例外
IOException | モデルファイルを開くときに例外が発生した場合。 |
---|
public Tensor getInputTensor (int inputIndex)
public Tensor getOutputTensor (int outputIndex)
public int[] getOutputTensorShape (int outputIndex)
出力シェイプを返します。グラフの作成時にのみ出力形状を決定する場合に便利です。
パラメータ
outputIndex |
---|
例外
IllegalStateException | インタープリタが閉じている場合に表示されます。 |
---|
public void run (Object[] 入力、Map<Integer、Object> の出力)
複数の入力に対してモデルの推論を実行し、複数の出力を返します。
パラメータ
入力 | 入力データの配列。入力はモデルの入力と同じ順序にする必要があります。各入力は、配列または多次元配列、または int、float、long、byte などのプリミティブ型の ByteBuffer です。サイズの大きい入力データを渡すには ByteBuffer をおすすめしますが、文字列型では(多次元の)配列入力パスを使用する必要があります。ByteBuffer を使用する場合、モデルの推論が完了するまで、その内容は変更されません。 |
---|---|
結果 | 出力インデックスを、出力データの多次元配列、または int、float、long、byte などのプリミティブ型の ByteBuffer にマッピングするマップ。使用する出力のエントリを保持する必要があるだけです。 |