编译 CoreML 模型
2023-06-26 16:54 作者:HyperAI超神经 | 我要投稿
本篇文章译自英文文档 Compile CoreML Models
作者是 Joshua Z. Zhang,Kazutaka Morita,Zhao Wu
更多 TVM 中文文档可访问 → https://tvm.hyper.ai/
本文介绍如何用 Relay 部署 CoreML 模型。
首先安装 coremltools 模块,可通过 pip 快速安装:
或参考官网:https://github.com/apple/coremltools
加载预训练的 CoreML 模型
这个例子使用 Apple 提供的预训练的 mobilenet 分类网络。
加载测试图像
还是用猫的图像:
在 Relay 上编译模型
现在应该对这个过程较为熟悉了。
输出结果:
在 TVM 上执行
这个过程与其他示例的相同。
查找分类集名称
在 1000 个类的分类集中,查找分数最高的第一个:
输出结果:
下载 Python 源代码:from_coreml.py
https://tvm.apache.org/docs/_downloads/3aeab7c9d659bf5da70126a1aff7c403/from_coreml.py
下载 Jupyter Notebook:from_coreml.ipynb
https://tvm.apache.org/docs/_downloads/a883b8474634054b6a79c17a288aa8ed/from_coreml.ipynb