編譯 CoreML 模型
2023-06-26 16:54 作者:HyperAI超神經(jīng) | 我要投稿
本篇文章譯自英文文檔?Compile CoreML Models?
作者是?Joshua Z. Zhang,Kazutaka Morita,Zhao Wu
更多 TVM 中文文檔可訪問 →?https://tvm.hyper.ai/
本文介紹如何用 Relay 部署 CoreML 模型。
首先安裝 coremltools 模塊,可通過 pip 快速安裝:
或參考官網(wǎng):https://github.com/apple/coremltools
加載預(yù)訓(xùn)練的 CoreML 模型
這個(gè)例子使用 Apple 提供的預(yù)訓(xùn)練的 mobilenet 分類網(wǎng)絡(luò)。
加載測試圖像
還是用貓的圖像:
在 Relay 上編譯模型
現(xiàn)在應(yīng)該對(duì)這個(gè)過程較為熟悉了。
輸出結(jié)果:
在 TVM 上執(zhí)行
這個(gè)過程與其他示例的相同。
查找分類集名稱
在 1000 個(gè)類的分類集中,查找分?jǐn)?shù)最高的第一個(gè):
輸出結(jié)果:
下載 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