创建一个 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()等查询方法