Cae*_*ael 2 machine-learning conv-neural-network keras tensorflow
我尝试使用以下代码将我的 CNN 模型 .h5 文件转换为 .tflite 文件:
import tensorflow as tf
from tf.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)
Run Code Online (Sandbox Code Playgroud)
但是然后有一个错误说:
ModuleNotFoundError: No module named 'tf'
Run Code Online (Sandbox Code Playgroud)
您不能从模块别名进行导入,您必须使用完整的模块名称:
from tensorflow.lite import TFLiteConverter
Run Code Online (Sandbox Code Playgroud)
你也可以直接tf.lite.TFLiteConverter在代码中引用
| 归档时间: |
|
| 查看次数: |
3814 次 |
| 最近记录: |