安装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 安装只在