Python 如何将.pb文件转换为.tflite。在tensorflow 2.1.0中

Python 如何将.pb文件转换为.tflite。在tensorflow 2.1.0中,python,tensorflow,deep-learning,computer-vision,tensorflow2.0,Python,Tensorflow,Deep Learning,Computer Vision,Tensorflow2.0,我已经使用SSD MobileNet V2 FPNLite 640x640和tensorflow 2.1.0对我的模型进行了培训。要创建.pb文件,请使用文件导出\u tflite\u graph\u tf2.py。现在,希望将.pb文件转换为.tflite。 我使用了tensorflow社区推荐的以下代码。但它正在将模型转换为1kb tflite文件 我也试着每晚使用tf,但同样的转换成1kb 将tensorflow导入为tf #转换模型 converter=tf.lite.TFLiteCon

我已经使用SSD MobileNet V2 FPNLite 640x640和tensorflow 2.1.0对我的模型进行了培训。要创建.pb文件,请使用文件导出\u tflite\u graph\u tf2.py。现在,希望将.pb文件转换为.tflite。 我使用了tensorflow社区推荐的以下代码。但它正在将模型转换为1kb tflite文件

我也试着每晚使用tf,但同样的转换成1kb

将tensorflow导入为tf
#转换模型
converter=tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)#到SavedModel目录的路径
tflite_model=converter.convert()
#保存模型。
将open('model.tflite','wb')作为f:
f、 写入(tflite_模型)

关于tf 1.x,同样的问题已经得到了回答。我尝试了这些代码。请尝试在选项中转换为tflite模型,并查看差异。谢谢!谢谢…tensorflow 2.1出现问题。。当尝试使用tf-2.3.0时,问题解决了。