跳到主要内容

创建一个 CoreML 视觉推理器 (coreml.new_vision_request)

声明

视觉推理器对象, 错误信息 = coreml.new_vision_request(编译好的模型路径)

视觉推理器对象, 错误信息 = coreml.new_vision_request({
compiled_model_path = 编译好的模型路径,
uses_cpu_only = 是否只使用 CPU,
compute_units = 计算单元配置,
image_crop_and_scale_option = 裁切与缩放模式,
})

参数

  • 编译好的模型路径
    文本型,目前支持图像分类和目标检测两种类型的模型

  • uses_cpu_only
    布尔型,可选参数,是否默认只使用 CPU 推理,默认 false

  • compute_units 文本型,可选参数,iOS 12+ 生效。支持以下值:

    • "all"
    • "cpu_only""cpu"
    • "cpu_and_gpu""gpu"
    • "cpu_and_neural_engine""ane""neural_engine"iOS 16+
  • image_crop_and_scale_option
    整数型,可选参数,视觉推理时的裁切与缩放模式

返回值

  • 视觉推理器对象
    视觉推理器对象,创建失败返回 nil

  • 错误信息
    文本型,创建成功为 nil,失败时返回错误信息

说明

  • 软件版本在 1.3.8 或以上方可使用
  • 不支持 iOS 12 以下版本系统
  • 这个函数用于加载一个已经编译好的模型 .mlmodelc 返回一个用于图像推理的上下文
  • 适用于以图像作为输入的 CoreML 视觉模型
  • 如果模型输入本身是通用 image / MLMultiArray 特征,也可考虑使用 coreml.new_model_request
  • 这个对象除了 :predict() / :run() 外,也支持 :results():is_done():compute_units():image_crop_and_scale_option():class_labels() 等查询方法

示例

参考 使用 CoreML 视觉推理器对图像进行推理