OSError: SavedModel文件不存在

Question

我正试图将我保存的模型转换为tflite模型，保存的模型保存在我的桌面上，但是当我尝试运行此代码时:生成错误-

OSError: SavedModel file does not exist at: C:/Users/Omar/Desktop/model00000014.h5/{saved_model.pbtxt|saved_model.pb}. 

我不知道是什么问题。

import tensorflow as tf
saved_model_dir = "r"C:/Users/Omar/Desktop/model00000014.h5""
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
tflite_model = converter.convert()
open("converted_model.tflite", "wb").write(tflite_model)

Answer 1

如果您试图将.h5Keras模型转换为TFLite模型，请确保使用TFLiteConverter.from_keras_model()方法，如文档中所述，

model = tf.keras.models.load( "C:/Users/Omar/Desktop/model00000014.h5" )
converter = tf.lite.TFLiteConverter.from_keras_model( model )
open( 'model.tflite' , 'wb' ).write( converter.convert() )

在SavedModel的情况下，使用TFLiteConverter.from_saved_model()并提及SavedModel目录的文件路径，

saved_model_dir = 'path/to/savedModelDir'
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)

您正在为TFLiteConverter.from_saved_model()方法提供Keras模型，这可能会导致错误。