OSError:保存的模型文件不存在tflite

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()方法，如docs中所述

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)

You're providing a Keras Model to the TFLiteConverter.from_saved_model() method, which might be causing an error.