我试图使用以下代码将Keras文件(.h5)中的模型转换为TensorFlow Lite文件(.tflite):
data = np.zeros((129, 326, 490, 3), dtype=np.uint8)
但是,以下行:
# Save model as .h5 keras file
keras_file = "eSleep.h5"
model_save = tf.keras.models.save_model(model,keras_file,overwrite=True,include_optimizer=True)
# Export keras file to TensorFlow Lite model
converter = tf.lite.TFLiteConverter.from_keras_model_file(keras_file)
tflite_model = converter.convert()
open("eSleep.tflite", "wb").write(tflite_model)
返回的错误:
tflite_model = converter.convert()
有人可以帮助我了解“图形中不存在初始化节点密集/内核/分配”的含义是什么以及如何解决该错误吗?
答案 0 :(得分:0)
夜间使用时,我也遇到同样的问题。我通过使用tensorflow 1.12解决了它。
答案 1 :(得分:0)
以我的经验,即使显示此错误,转换后的模型也可以正常工作。您可以忽略该错误。