欢迎光临散文网 会员登陆 & 注册

编译 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


编译 CoreML 模型的评论 (共 条)

分享到微博请遵守国家法律