java调用onnx模型_ONNX预训练模型加载

tvm官网中，对从ONNX预训练模型中加载模型的教程说明

教程来自于：https://docs.tvm.ai/tutorials/frontend/from_onnx.html#sphx-glr-tutorials-frontend-from-onnx-py

首先我对教程进行了一些修改，很多东西没有必要，比如不是每次都需要从网上下载图片和模型，super_resolution.onnx和cat.png都预先下载到了文件同目录下，

同时，最新版本的tvm中不支持Python2.7，我没有编译llvm，所以我把我的设置都改到了cuda上，在24行和32行有体现,注意最新版本

1 importonnx2 importnumpy as np3 importtvm4 importtvm.relay as relay5 #from tvm.contrib.download import download_testdata

6

7 #model_url = ''.join(['https://gist.github.com/zhreshold/',

8 #'bcda4716699ac97ea44f791c24310193/raw/',

9 #'93672b029103648953c4e5ad3ac3aadf346a4cdc/',

10 #'super_resolution_0.2.onnx'])

11 #model_path = download_testdata(model_url, 'super_resolution.onnx', module='onnx')

12 #now you have super_resolution.onnx on