如何与扩展共享CoreML模型？

Question

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

Answer 1

您必须在您的项目中创建一个共享框架，该框架也包含在应用程序和扩展中。你把模型放在框架的资源文件夹中，你可以在框架的bundle上为模型创建路径或URL。

编辑：如果您将生成的模型类放置在框架中，您生成的模型应自动找到模型数据。它创建从类捆的模型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!) 
} 

在从骨架的类的情况下，Bundle(for: MarsHabitatPricer.self)是框架的束。您的框架只需要导出模型的类，以便您可以在应用程序和扩展中使用它。

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