使用InceptionV3和Food101创建TFLite文件

Question

我正在尝试创建要在Android上使用的TFLITE模型。我将InceptionV3Model与Food101数据集一起使用。我是这门机器学习世界的新手。我从TensorFlow中找到了我试图使用的代码：

https://github.com/tensorflow/hub/blob/master/examples/image_retraining/retrain.py

我不知道将模型转换为TFLite的位置以及添加的内容。

请向我解释检查点和冻结图，并帮助创建TFLite模型。

谢谢。

Answer 1

您可以先通过设置--saved_model_dir将模型导出到saved_model然后使用TFLiteConverter.from_saved_model。

https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/lite/TFLiteConverter#from_saved_model