Как я могу протестировать модель .tflite, чтобы доказать, что она ведет себя как исходная модель, используя те же данные испытаний?

Question

Я сгенерировал модель .tflite на основе обученной модели. Я хотел бы проверить, что модель tfilte дает те же результаты, что и исходная модель.

Предоставление одинаковых данных испытаний и получение одинаковыхрезультат.

Answer 1

В дополнение к ответу, данному @ miaout17, для отладки / понимания вашей модели tflite (что является духом вопроса) вы можете

• использовать --dump_graphviz для визуализации графика какздесь объяснено https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/lite/toco/g3doc/cmdline_examples.md#using---dump_graphviz
• используйте flatc для генерации API Python и затем проанализируйте модель с помощью этого API https://google.github.io/flatbuffers/flatbuffers_guide_use_python.html
• генерируйте json из файла tflite с использованием flatc и распечатайте его
• Используйте визуализатор tflite: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/lite/tools/visualize.py (в настоящее время имеется ошибка )

Answer 2

Вы можете использовать интерпретатор TensorFlow Lite Python для проверки вашей модели tflite.

Позволяет вам вводить входные данные в оболочке Python и читать результаты напрямую, как если бы вы использовали обычную модель тензорного потока.

Я ответил на этот вопрос здесь .

И вы можете прочитать это Официальное руководство TensorFlow lite для получения подробной информации.

Я также нашел очень хороший инструмент визуализации , который может напрямую загружать файл .tflite, чтобы вы могли проверить архитектуру вашей модели и ее вес.

Answer 3

В базе кода TensorFlow есть tflite_diff_example_test .Он генерирует случайные данные и передает те же данные в TensorFlow & TensorFlow lite, а затем сравнивает, находится ли разница в пределах небольшого порога.

Вы можете извлечь код TensorFlow из Github и запустить его с помощью bazel:

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

, тогда вы увидите, какие аргументы вам нужно передать.