不能导入名称为'TFLiteConverter'?

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

我使用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'

我指的是这个 文件

python-3.x tensorflow tensorflow-lite
1个回答
0
投票
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)

这应该可以工作。

© www.soinside.com 2019 - 2024. All rights reserved.