不经优化的模型
import onnx
from tvm.contrib.download import download_testdata
from PIL import Image
import numpy as np
import tvm.relay as relay
import tvm
from tvm.contrib import graph_executor
加载 ONNX 模型
"""
model_url = "".join(
[
"https://github.com/onnx/models/tree/",
"main/vision/classification/resnet/model/",
"resnet50-v2-7.onnx",
]
)
model_path = download_testdata(model_url, "resnet50-v2-7.onnx", module="onnx")
onnx_model = onnx.load(model_path)
"""
# 使用提前下载的模型
onnx_model = onnx.load("./resnet50-v2-7.onnx")
加载测试图和预处理
"""
img_url = "https://s3.amazonaws.com/model-server/inputs/kitten.jpg"
img_path = download_testdata(img_url, "imagenet_cat.png", module="data")
"
