如何与扩展共享 CoreML 模型？

Question

我正在与应用程序扩展共享 CoreML 模型。 Xcode 为模型生成 swift 代码。为了在扩展中使用该代码，我还必须在应用程序扩展中包含相同的模型。有没有办法在应用程序中只包含一次模型，并在应用程序和扩展程序中使用？

Answer 1

您必须在您的项目中创建一个共享框架，该框架也包含在应用程序和扩展程序中。您将模型放在框架的资源文件夹中，然后可以通过框架的包创建模型的路径或 url。

编辑：如果您将生成的模型 class 也放在框架中，您生成的模型应该会自动找到模型数据。它从 class 的包中创建模型 url（参见生成的 init 方法）：

init(contentsOf url: URL) throws {
    self.model = try MLModel(contentsOf: url)
}
convenience override init() {
    let bundle = Bundle(for: MyModel.self)
    let assetPath = bundle.url(forResource: "MyModel", withExtension:"mlmodelc")
    try! self.init(contentsOf: assetPath!)
}

对于框架中的 class，Bundle(for: MarsHabitatPricer.self) 是框架的捆绑包。您的框架只需要导出模型的 class，这样您就可以在应用程序和扩展程序中使用它。

另请注意，使用同样生成的 init(contentsOf:)，您可以从 everywere 加载您喜欢的模型。

如何与扩展共享 CoreML 模型？

How to share CoreML model with extension?

ios

coreml