Core ML支持很多机器学习的模型,包括神经网络,决策树融合,支持向量机,线性模型。CoreML需要一个扩展名为 .mlmodel的机器学习模型。苹果提供了很多开源的机器学习模型https://developer.apple.com/machine-learning。
当然你也可以将你自己的模型转换为苹果所支持的机器学习模型格式。苹果提供了Core ML Tool用来将你自己的机器学习模型转换成苹果支持的机器学习模型格式。当然,暂时只支持如下表所示的机器学习框架生成的模型。
根据你所用的第三方工具来选择不同的转换器,将第三方工具生成的模型转换为苹果支持的模型格式。例如,如果你用Caffe生成了Caffe模型,它的扩展名为.caffemodel,那么代码看上去应该是这样的:
import coremltools
coreml_model = coremltools.converters.caffe.convert('my_caffe_model.caffemodel')
coreml_model.save('my_model.mlmodel')
当然,根据你所使用的第三方工具不同,你可能还需要对生成的模型进行一些必要的修改,例如输入,输出,训练数据标签,或者你需要定义图形的名称,格式等等。可以通过阅读工具附带的文档来完成这些修改。
你也可以自己写一个模型转换工具,将第三方模型转换为苹果支持的模型格式,详细请参照苹果机器学习模型定义文档。
网友评论