本篇文章译自英文文档 Compile Keras Models

作者是 Yuwei Hu

更多 TVM 中文文档可访问 →TVM 中文站。https://tvm.hyper.ai/ 

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

首先安装 Keras 和 TensorFlow，可通过 pip 快速安装：

或参考官网：https://keras.io/#installation

加载预训练的 Keras 模型

加载 Keras 提供的预训练 resnet-50 分类模型：

加载测试图像

这里使用的还是先前猫咪的图像：

输出结果：

使用 Relay 编译模型

将 Keras 模型（NHWC 布局）转换为 Relay 格式（NCHW 布局）：

在 TVM 上执行

查找分类集名称

在 1000 个类的分类集中，查找分数最高的第一个：

输出结果：

下载 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