【tfcoreml】tensorflow向CoreML模型的转换工具封装

安装tf向apple coreml模型转换包tfcoreml
基于苹果自己的转换工具coremltools进行封装

tfcoreml

为了将训练的模型转换到apple中使用,需要将模型转换为ios支持的mlmodel形式。目前苹果官方的推荐使用Core ML tools来进行转换。
它支持

Scikit Learn
LIBSVM
Caffe
Keras
XGBoost.
Tensorflow
MXNet

等工具包的模型转换。
但为了更方便的使用tensorflow的模型,在github中找到了tfcoreml的封装,直接对tensorflow的pb模型进行转换。

首选需要安装tfcoreml的依赖:

tensorflow >= 1.5.0
coremltools >= 0.8
numpy >= 1.6.2
protobuf >= 3.1.0
six >= 1.10.0

在安装CoreMLtools时会遇到以下问题:

  • 1.coremltools的pip 安装只在
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值