如何将样式转移 tensorflow 模型转换为具有灵活输入形状的 mlmodel？

Question

我已阅读 Coreml 指南，该指南展示了如何使用 coremltools 将 pb 模型转换为 mlmodel。但是，在尝试遵循指南时出现以下错误。这意味着输入形状必须是特定的。

ValueError: "ResizeBilinear" op: 第二个输入，即输出大小，必须是静态已知的

那么，有谁知道如何转换灵活的输入形状mlmodel？

这是我的代码：

import coremltools as ct

def mlmodel_image(pb):
    input_shape = ct.Shape(shape=(1, ct.RangeDim(1, 720), ct.RangeDim(1, 1280), 3))
    model_input = ct.ImageType(shape=input_shape)
    mlmodel = ct.convert(pb, inputs=[model_input], source='TensorFlow')
    mlmodel.save(pb.replace(".pb", "_img.mlmodel"))
    print('------save to ', pb.replace(".pb", "_img.mlmodel"))

Answer 1

请尝试我的示例： https://github.com/dhrebeniuk/RealTimeFastStyleTransfer

并查看我在 PyTorch 中附带 Google Colab Notebook 的文章。

有说明如何在 iOS 上以最佳性能运行 Style Transfer。