TVM编译ONNX模型

编译ONNX模型

  • 加载预训练的ONNX模型
  • 加载测试图像
  • 用Relay编译模型
  • 在TVM上执行
  • 显示结果

本文是介绍如何使用Relay部署ONNX模型的入门教程。
首先,必须安装ONNX软件包。
一个快速的解决方案是安装protobuf编译器,然后

pip install onnx --user

或请参考官方网站。 https://github.com/onnx/onnx

import onnx
import numpy as np
import tvm
from tvm import te
import tvm.relay as relay
from tvm.contrib.download import download_testdata

加载预训练的ONNX模型

这里使用的示例超分辨率模型与onnx教程http://pytorch.org/tutorials/advanced/super_resolution_with_caffe2.html中的模型完全相同, 我们跳过了pytorch模型构建部分,并下载了保存的onnx模型

model_url="".join(
    [
        "https://gist.github.com/zhreshold/",
        "bcda4716699a

你可能感兴趣的:(TVM)