使用InceptionV3和Food101创建TFLite文件

问题描述 投票:0回答:1

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

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

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

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

谢谢。

python machine-learning image-recognition
1个回答
0
投票

您可以先通过设置--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

热门问题
推荐问题
最新问题