使用coremltools转换张量流检测模型

时间:2020-09-10 07:45:04

标签: tensorflow object-detection coremltools

我已经使用本教程创建了一个检测模型。 https://tensorflow-object-detection-api-tutorial.readthedocs.io/en/latest/training.html#

我可以使用coreml工具将其转换吗?

当我使用tf.saved_model.load时,我得到一个

<tensorflow.python.saved_model.load.Loader._recreate_base_user_object.<locals>._UserObject object at 0x000001C86252F3D0>.

因此coremltools返回此错误。

NotImplementedError:预期的模型格式:[SavedModel | [具体功能] | tf.keras.Model | .h5],

这个对象是什么?

谢谢!

代码示例:

new_model = tf.saved_model.load("my-model/saved_model")
image_input = ct.ImageType(shape=(1, 224, 224, 3,),
                          bias=[-1,-1,-1], scale=1/127)
core_ml_model = ct.convert( new_model, inputs=[image_input], source="tensorflow")
core_ml_model.save("SSDMobileNetV2.mlmodel")

ps:我也尝试过 core_ml_model = ct.convert("my-model/saved_model") 哪个给

TypeError:“ Var”对象不可下标

0 个答案:

没有答案