转换代码如下:
import tensorflow as tf
model = tf.keras.models.load_model("action.h5")
converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_model = converter.convert()
open("action.tflite", "wb").write(tflite_model)
tensorflow版本是2.6.0,keras版本是2.9.0
运行之后报错
AttributeError: module 'keras.api._v2.keras' has no attribute 'models'
百度了半天,后来有同学说是版本的问题,把keras版本降到了2.6.0,再次运行无错误。
.tflite文件生成成功