编译ONNX模型
下面描述使用Relay如何部署ONNX模型:
#安装onnx,https://github.com/onnx/onnx
pip install onnx --user
首先导入所需要的python包:
import onnx
import numpy as np
import tvm
import tvm.relay as relay
from tvm.contrib.download import download_testdata
加载预训练ONNX模型
model_url = ''.join(['https://gist.github.com/zhreshold/', 'bcda4716699ac97ea44f791c24310193/raw/','93672b029103648953c4e5ad3ac3aadf346a4cdc/','super_resolution_0.2.onnx'])
model_path = download_testdata(model_url, 'super_resolution.onnx', module='onnx')
#从硬盘加载supe_resolution.onnx
onnx_model = onnx.load(model_path)
加载一张测试图像
fro