Core ML(基础) - 转换训练模型为Core ML

一、概述

如果您的模型是被支持的第三方机器学习框架创建和训练的,您可以通过Core ML 的工具或者第三方的转换工具,比如MXNet Converter或者TensorFlow converter,使用这些工具来转换您的模型为Core ML格式。否则,您需要创建您自己的转换工具。

二、使用Core ML工具

Core ML Tools是一个Python包,它可以将不同类型的Model转换为Core ML格式的模型。下表列出了Core Ml Tools支持的模型和第三方框架。

Model type Supported models Supported frameworks
Neural networks Feedforward, convolutional, recurrent Caffe v1
Keras 1.2.2+
Tree ensembles Random forests, boosted trees, decision trees scikit-learn 0.18
XGBoost 0.6
Support vector machines Scalar regression, multiclass classification scikit-learn 0.18
LIBSVM 3.22
Generalized linear models Linear regression, logistic regression scikit-learn 0.18
Feature engineering Sparse vectorization, dense vectorization, categorical processing scikit-learn 0.18
Pipeline models Sequentially chained models scikit-learn 0.18

三、转换模型

使用与模型的第三方框架相应的Core ML转换器来转换您的模型。调用转换器的转换方法并保存转换完成的Model保存为Core ML模型格式(.mlmodel)。
比如,如果您的模型是使用Caffe创建的,那么将Caffe model (.caffemodel) 传入到coremltools.converters.caffe.convert方法。

import coremltools
coreml_model = coremltools.converters.caffe.convert('my_caffe_model.caffemodel')

然后将的到的模型保存为Core ML模型格式。

coremltools.utils.save_spec(coreml_model, 'my_model.mlmodel')

依据您的模型,您可能需要更新输入、输出和标签,或者您可能需要生命图片的名字、类型和格式。转换工具和更多的文件绑定在一起,因为可用的选项因工具而异。关于Core ML Tools更多的信息,请参阅Package Documentation

四、编写自己的转换工具

当您需要转换一个不包含在上边列表中的支持的格式类型时,可以创建自己的转换工具。
编写您自己的转换工具涉及到将您的模型的输入、输出和体系结构的表示转换为Core ML模型格式。您可以通过定义模型架构的每一层及它与其他层的连接来实现这一点。以Core ML Tools提供的转换工具为例,他们演示了如何将从第三方框架创建的各种模型类型转换为Core ML模型格式。

注:
Core ML 模型类型是由一组协议缓冲区文件定义,并在Core ML Model Specification中有详细描述。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容