我已经使用本教程创建了一个检测模型。 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”对象不可下标