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

后端 未结 3 1421
孤城傲影
孤城傲影 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:12

    In addition to the answer given by @miaout17, to debug / understand your tflite model (which is the spirit of the question), you can

    • use --dump_graphviz to visualize the graph as explained here https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/lite/toco/g3doc/cmdline_examples.md#using---dump_graphviz
    • use flatc to generate a python api and then parse the model via that api https://google.github.io/flatbuffers/flatbuffers_guide_use_python.html
    • generate json from the tflite file using flatc and print it out
    • Use the tflite visualizer: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/lite/tools/visualize.py (currently has a bug)

提交回复
热议问题