How can I test a .tflite model to prove that it behaves as the original model using the same Test Data?

后端 未结 3 1419
孤城傲影
孤城傲影 2020-12-15 17:53

I have generated a .tflite model based on a trained model, I would like to test that the tfilte model gives the same results as the original model.

Giving both the s

3条回答
  •  一整个雨季
    2020-12-15 18:28

    There is a tflite_diff_example_test in the TensorFlow code base. It generates random data and feed the same data into TensorFlow & TensorFlow lite, then compare if the difference is within a small threshold.

    You can to checkout TensorFlow code from Github, and run it with bazel:

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

    then you'll see what arguments you need to pass.

提交回复
热议问题