編譯 Keras 模型
2023-06-12 11:55 作者:HyperAI超神經(jīng) | 我要投稿
本篇文章譯自英文文檔?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
