我想在android studio中使用我的keras训练模型。我在互联网上得到了这段代码,将我的代码从keras转换为tensorflow-lite。但是当我尝试编写代码时,我得到了这个错误:
OSError: SavedModel file does not exist at: C:\Users\Munib\New folder/{saved_model.pbtxt|saved_model.pb}
我使用的代码是从keras转换到tensorflow-lite:
import tensorflow as tf
# Converting a SavedModel to a TensorFlow Lite model.
saved_model_dir = r"C:\Users\Munib\New folder"
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
tflite_model = converter.convert()
# Converting a tf.Keras model to a TensorFlow Lite model.
converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_model = converter.convert()
# Converting ConcreteFunctions to a TensorFlow Lite model.
converter = tf.lite.TFLiteConverter.from_concrete_functions([func])
tflite_model = converter.convert()
发布于 2020-06-13 19:59:34
首先,最好使用相对路径而不是绝对路径。其次,如果您使用model.save('my_model')
,则keras将为您创建一个名为my_model
的目录,您应该在该目录中找到一个扩展名为pb
或pbtxt
的文件,这是您应该用于tflite转换器的目录
发布于 2021-02-18 10:22:07
我在使用google Colab的时候也遇到了这个问题。似乎,我的Colab丢失了模型所在的目录路径数据。只需再次运行这两行代码,然后重试。希望这能解决这个问题:
import os
os.chdir("/content/drive/My Drive/path/to/your/model")
https://stackoverflow.com/questions/62358745
复制相似问题