TVMコンパイルONNXモデル

824 ワード
TVM
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_Caffe 2.htmlのモデルはまったく同じで、pytorchモデル構築部分をスキップし、保存したonnxモデルをダウンロードしました.
    model_url="".join(
    [
        "https://gist.github.com/zhreshold/",
        "bcda4716699a
    ubuntuシステムにおけるdockerのインストール
    Spring Boot+Spring Data JPA+Spring Cache