TVM编译MXNet模型

本文是介绍如何使用Relay部署mxnet模型的入门教程。
首先，需要安装mxnet模块。
一个快速的解决方案是

pip install mxnet --user

或请参阅官方安装指南。 https://mxnet.apache.org/versions/master/install/index.html

import mxnet as mx
import tvm
import tvm.relay as relay
import numpy as np

从Gluon Model Zoo下载Resnet18模型

在本节中，我们下载预训练的imagenet模型并对图像进行分类。

from tvm.contrib.download import download_testdata
from mxnet.gluon.model_zoo.vision import get_model
from PIL import Image
from matplotlib import pyplot as plt\

block=get_model("resnet18