使用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,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”对象不可下标

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）