尝试将 Keras .h5 文件转换为 .tflite 文件时：转换器给出 AttributeError: 'str' object has no attribute 'call'

Question

“tflite_model = converter.convert()”行给出了 AttributeError: 'str' object has no attribute 'call'。

查看代码截图->1

代码：

import tensorflow as tf
converter = tf.lite.TFLiteConverter.from_keras_model('///Users/theunskuhn/Desktop/Savedfile/basic_malaria_pos_neg_v3.h5')
converter.experimental_new_converter = True
tflite_model = converter.convert()
open("basic_malaria_pos_neg_v3.tflite", "wb").write(tflite_model)  

错误： AttributeError: 'str' 对象没有属性 'call'

错误指向第 4 行：“tflite_model = converter.convert()”

Screenshot of new code from answer below

Answer 1

如果您在 TensorFlow 2.0 或更高版本中使用 TFLiteConverter API，TFLiteConverter.from_keras_model 会接受 Keras Model 对象，而不是模型的路径 str。

首先，使用tf.keras.models.load_model() 加载模型，然后将此模型传递给TFLiteConverter API。

import tensorflow as tf

model = tf.keras.models.load_model( '///Users/theunskuhn/Desktop/Savedfile/basic_malaria_pos_neg_v3.h5' )

converter = tf.lite.TFLiteConverter.from_keras_model( model )
tflite_model = converter.convert()
open("basic_malaria_pos_neg_v3.tflite", "wb").write(tflite_model)  

方法 TFLiteConverter.from_keras_model_file() 在 TF 2.0 中被 TFLiteConverter.from_keras_model() 取代。请参阅docs。