本篇文章譯自英文文檔?Compile Keras Models

作者是?Yuwei Hu

更多 TVM 中文文檔可訪問 →TVM 中文站。https://tvm.hyper.ai/?

本文介紹如何用 Relay 部署 Keras 模型。

首先安裝 Keras 和 TensorFlow，可通過 pip 快速安裝：

或參考官網(wǎng)：https://keras.io/#installation

加載預(yù)訓練的 Keras 模型

加載 Keras 提供的預(yù)訓練 resnet-50 分類模型：

加載測試圖像

這里使用的還是先前貓咪的圖像：

輸出結(jié)果：

使用 Relay 編譯模型

將 Keras 模型（NHWC 布局）轉(zhuǎn)換為 Relay 格式（NCHW 布局）：

在 TVM 上執(zhí)行

查找分類集名稱

在 1000 個類的分類集中，查找分數(shù)最高的第一個：

輸出結(jié)果：

下載 Python 源代碼：from_keras.py

https://tvm.apache.org/docs/_downloads/c23f7654585d9b0fa2129e1765b2a8f2/from_keras.py

下載 Jupyter Notebook：from_keras.ipynb?

https://tvm.apache.org/docs/_downloads/c82f632d47458e76d2af9821b6778e36/from_keras.ipynb