将保存的tensorflow模型转换为tensorflow Lite的正确方法是啥
Posted
技术标签:
【中文标题】将保存的tensorflow模型转换为tensorflow Lite的正确方法是啥【英文标题】:What is the right way to convert saved tensorflow model to tensorflow Lite将保存的tensorflow模型转换为tensorflow Lite的正确方法是什么 【发布时间】:2020-11-29 04:10:09 【问题描述】:我有一个保存的 tensorflow 模型,与 model zoo 中的所有模型相同。
我想把它转换成tesorflow lite,我从tensorflow github上找到了以下方法(我的tensorflw版本是2):
!wget http://download.tensorflow.org/models/object_detection/tf2/20200711/ssd_resnet50_v1_fpn_640x640_coco17_tpu-8.tar.gz
# extract the downloaded file
!tar -xzvf ssd_resnet50_v1_fpn_640x640_coco17_tpu-8.tar.gz
!pip install tf-nightly
import tensorflow as tf
converter = tf.lite.TFLiteConverter.from_saved_model('ssd_mobilenet_v2_320x320_coco17_tpu-8/saved_model')
converter.optimizations = [tf.lite.Optimize.DEFAULT]
converter.experimental_new_converter = True
converter.target_spec.supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS, tf.lite.OpsSet.SELECT_TF_OPS]
tflite_model = converter.convert()
open("m.tflite", "wb").write(tflite_model)
但是转换后的模型的输出和输入形状与原始模型不匹配,请检查以下内容:
原始模型输入和输出形状 转换后的模型输入和输出形状所以这里有问题!输入/输出形状应与原始模型匹配! 有什么想法吗?
【问题讨论】:
【参考方案1】:从 Tensorflow github 问题中,我使用了他们的答案来解决我的问题。 Link
他们的方法:
!pip install tf-nightly
import tensorflow as tf
## TFLite Conversion
model = tf.saved_model.load("saved_model")
concrete_func = model.signatures[tf.saved_model.DEFAULT_SERVING_SIGNATURE_DEF_KEY]
concrete_func.inputs[0].set_shape([1, 300, 300, 3])
tf.saved_model.save(model, "saved_model_updated", signatures="serving_default":concrete_func)
converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir='saved_model_updated', signature_keys=['serving_default'])
converter.optimizations = [tf.lite.Optimize.DEFAULT]
converter.target_spec.supported_ops = [tf.lite.OpsSet.TFLITE_BUILTINS, tf.lite.OpsSet.SELECT_TF_OPS]
tflite_model = converter.convert()
## TFLite Interpreter to check input shape
interpreter = tf.lite.Interpreter(model_content=tflite_model)
interpreter.allocate_tensors()
# Get input and output tensors.
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()
# Test the model on random input data.
input_shape = input_details[0]['shape']
print(input_shape)
[1 300 300 3]
谢谢MeghnaNatraj
【讨论】:
当interpreter.invoke()
被调用时会发生什么?当我使用以这种方式创建的tflite
模型时,我在interpreter
上调用invoke()
后得到RuntimeError: Container __per_step_0 does not exist. (Could not find resource: __per_step_0/_tensor_arraysTensorArrayV3_0) (while executing 'TensorArrayScatterV3' via Eager)Node number 237 (TfLiteFlexDelegate) failed to invoke.
。尽管tflite
模型(来自tensorflow.org/lite/guide/hosted_models)已经可用,但不要抛出此错误。我已经转换了一个在自定义数据上训练的 ssd_mobilenet_v2_coco_2018_03_29
模型。【参考方案2】:
两个模型输入输出的形状应该如下图一样
如果模型已经是 saved_model 格式,你下面的代码
# if you are using same model
export_dir = 'ssd_mobilenet_v2_320x320_coco17_tpu-8/saved_model'
converter = tf.lite.TFLiteConverter.from_saved_model(export_dir)
如果您的模型是 Keras 格式,请使用以下格式
# if it's a keras model
model = tf.keras.applications.MobileNetV2(weights="imagenet", input_shape= (224, 224, 3))
converter = tf.lite.TFLiteConverter.from_keras_model(model)
在这两种情况下,目的都是为了得到转换器。
我没有saved_model,所以我将使用keras模型并将其转换为saved_model格式,以Keras模型格式为例
import pathlib #to use path
model = tf.keras.applications.MobileNetV2(weights="imagenet", input_shape= (224, 224, 3))
export_dir = 'imagenet/saved_model'
tf.saved_model.save(model, export_dir) #convert keras to saved model
converter = tf.lite.TFLiteConverter.from_saved_model(export_dir)
converter.optimizations = [tf.lite.Optimize.DEFAULT] #you can also optimize for size or latency OPTIMIZE_FOR_SIZE, OPTIMIZE_FOR_LATENCY
tflite_model = converter.convert()
#save the model
tflite_model_file = pathlib.Path('m.tflite')
tflite_model_file.write_bytes(tflite_model)
tflite_interpreter = tf.lite.Interpreter(model_path= 'm.tflite') #you can load the content with model_content=tflite_model
# get shape of tflite input and output
input_details = tflite_interpreter.get_input_details()
output_details = tflite_interpreter.get_output_details()
print("Input: ".format( input_details[0]['shape']))
print("Output:".format(output_details[0]['shape']))
# get shape of the origin model
print("Input: ".format( model.input.shape))
print("Output: ".format(model.output.shape))
对于 tflite:我有这个
对于原始模型,我有这个
您会看到tflite
和keras
模型的形状相同
【讨论】:
我的模型是 saved_model.pb 而不是 model.h5,所以它不是 keras 模型 编辑我的问题并添加模型链接,请确认是否有匹配的输入输出download.tensorflow.org/models/object_detection/tf2/20200711/… @H.H saved_model 似乎有问题。我得到了这个None is only supported in the 1st dimension. Tensor 'input_tensor' has invalid shape '[1, None, None, 3]'
我尝试了另一个测试保存的模型,它工作正常
所以我更新了代码,并将 Keras 模型转换为 saved_model 并使用了 saved_model 版本。它仍然按预期工作。所以你使用的下载的 saved_model 格式有问题
你可以试试这个型号吗download.tensorflow.org/models/object_detection/tf2/20200711/…【参考方案3】:
只需重塑你的输入张量。
你可以使用resize_tensor_input
函数,像这样:
interpreter.resize_tensor_input(input_index=0, tensor_size=[1, 640, 640, 3])
现在您输入的形状将是:[1, 640, 640, 3]
。
【讨论】:
以上是关于将保存的tensorflow模型转换为tensorflow Lite的正确方法是啥的主要内容,如果未能解决你的问题,请参考以下文章
将 tf.Tensor 转换为 numpy 数组,然后将其保存为图像而不使用 eager_execution
将 keras 模型从 pb 文件转换为 tflite 文件