我想加载一个我已经训练并保存为.pb格式的Keras模型。
以下是代码,
我使用的是Jupyter Notebook。
模型成功保存为saved_model.pb,位于同一目录下。但是代码无法访问它。
有谁能帮我看看,如何访问保存为.pb扩展名的这个Keras模型?
我在其他几个地方寻找解决方案,但没有找到。
模型保存路径为model/saved_model.pb。
我已经将.pb文件取出并放置在代码文件所在的同一目录下。
回答:
函数tf.keras.models.load_model
可以将SavedModel
加载为tf.keras
模型。该函数的参数是保存模型的路径。
所以尝试使用model = tf.keras.models.load_model('model')