web-dev-qa-db-ja.com

.tfliteモデルをテストして、同じテストデータを使用して元のモデルとして動作することを証明するにはどうすればよいですか?

訓練されたモデルに基づいて.tfliteモデルを生成しました。tfilteモデルが元のモデルと同じ結果を与えることをテストしたいと思います。

同じテストデータを提供し、同じ結果を取得します。

16
Jorge Jiménez

TensorFlow Lite Pythonインタープリターを使用してtfliteをテストできますモデル。

python Shellで入力データをフィードし、通常のテンソルフローモデルを使用しているように出力を直接読み取ることができます。

私はこの質問に答えました here

そして、あなたはこれを読むことができます TensorFlow lite公式ガイド 詳細情報.

また、 非常に優れた視覚化ツール を見つけました。これは.tfliteファイルを直接ロードして、モデルアーキテクチャとモデルの重みを検査できます。

12
Jing Zhao

TensorFlowコードベースには tflite_diff_example_test があります。ランダムデータを生成し、同じデータをTensorFlowとTensorFlow liteにフィードし、差が小さなしきい値内にあるかどうかを比較します。

GithubからTensorFlowコードをチェックアウトし、bazelで実行できます。

bazel run //tensorflow/contrib/lite/testing:tflite_diff_example_test

その後、どの引数を渡す必要があるかがわかります。

2
miaout17

@ miaout17による回答に加えて、tfliteモデルをデバッグ/理解するために(質問の精神です)、次のことができます。

1
Pannag Sanketi