Swift加载自定义CoreML模型编译及运行失败问题求助

根因说明 #

你遇到的报错属于推荐类CoreML模型的常见问题，模型在CreateML中运行正常说明模型结构本身没有逻辑问题，报错集中在文件读取/编译产物解析环节，可按以下步骤逐一排查：

排查步骤 #

• 优先排查沙箱/文件权限问题
  macOS/iOS的沙箱机制默认禁止应用读取沙箱外的文件，你使用桌面绝对路径读取模型会被系统拦截，即使手动赋予权限也容易出现偶发读取失败。请将.mlmodel文件移入项目的Resources目录，在Xcode的Build Phases → Copy Bundle Resources中确认模型已被添加，使用Bundle相对路径读取：

  guard let modelURL = Bundle.main.url(forResource: "ItemSimilarityRecommender", withExtension: "mlmodel") else {
      fatalError("项目内未找到目标模型文件")
  }
  do {
      let compiledURL = try MLModel.compileModel(at: modelURL)
      let model = try MLModel(contentsOf: compiledURL)
  } catch {
      print(error.localizedDescription)
  }
  

  如果你是直接使用预编译的.mlmodelc文件，需要将整个.mlmodelc文件夹作为目录添加到项目的Copy Bundle Resources中，不要单独添加内部的零散文件。

• 检查CoreML版本与部署系统的兼容性
  CreateML默认会导出适配最新系统的CoreML规格版本，高版本规格的模型在低版本系统上会出现protobuf解析失败、coremldata.bin无效的误报：

  1. 回到CreateML导出模型的环节，选择「导出」而非直接存储
  2. 在导出选项的「CoreML版本」下拉栏，选择与你项目最低部署版本匹配的规格：
     • 最低部署iOS13+/macOS10.15+ → 选CoreML 3
     • 最低部署iOS14+/macOS11+ → 选CoreML 4
     • 最低部署iOS15+/macOS12+ → 选CoreML 5
  3. 重新导出后替换项目内的旧模型即可。

• 验证模型文件完整性
  如果上述步骤无效，可通过苹果官方的coremltools工具验证模型文件是否存在隐性损坏：

  1. 安装coremltools：pip install coremltools
  2. 运行验证脚本：

  import coremltools as ct
  # 替换为你的模型文件路径
  model = ct.models.MLModel("./ItemSimilarityRecommender.mlmodel")
  # 输出模型元信息，无报错则说明模型结构正常
  print(model.spec.description)
  

  如果脚本运行报错，说明模型在导出环节出现了结构损坏，回到CreateML重新训练后再导出即可。

内容的提问来源于stack exchange，提问作者Ben