ImageClassifier

画像の分類を実行します。

API は、オプションではあるが強く推奨される TFLiteモデル メタデータを備えた TFLite モデルを想定しています。 。

API は、1 つの画像入力テンソルと 1 つの分類出力テンソルを持つモデルをサポートします。より具体的に言うと、次のような要件があります。

• 入力画像テンソル ( kTfLiteUInt8 / kTfLiteFloat32 )
  • サイズ[batch x height x width x channels]の画像入力。
  • バッチ推論はサポートされていません ( batch 1 である必要があります)。
  • RGB 入力のみがサポートされています ( channels 3 である必要があります)。
  • type がkTfLiteFloat32の場合、入力正規化のために NormalizationOptions をメタデータに付加する必要があります。
• 出力スコア テンソル ( kTfLiteUInt8 / kTfLiteFloat32 )
  • [1 x N]や[1 x 1 x 1 x N]など、2 次元または 4 次元のNクラスを使用します。
  • ラベル ファイルをメタデータにパックする必要があります。画像分類器のメタデータの作成例を参照してください。ラベル ファイルがパックされていない場合は、結果のラベルとしてインデックスが使用されます。

このようなモデルの例はTensorFlow Hub にあります。 。