我使用Colab (Tensorflow 1.15.0版本,Python 3笔记本)使用以下代码导入TFLiteConverter。
from tensorflow.lite import TFLiteConverter
converter = lite.TFLiteConverter.from_saved_model('/drive/My Drive/FSD_modelV09A.h5')
tflite_model = converter.convert()
open("/drive/My Drive/FSD_modelV09A.tflite", "wb").write(tflite_model)
然后发生了这个错误。
ImportError: cannot import name 'TFLiteConverter'
我指的是这个 文件
from tensorflow import lite
converter = lite.TFLiteConverter.from_saved_model('/drive/My Drive/FSD_modelV09A.h5')
tflite_model = converter.convert()
open("/drive/My Drive/FSD_modelV09A.tflite", "wb").write(tflite_model)
这应该可以工作。