.h5文件转.tflite文件(tensorflow2.6)

转换代码如下:

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文件生成成功

你可能感兴趣的:(Tensorflow,tensorflow,tflite)