Chainer-NNVM-Example

Chainer Advent Calendar 2017 - Qiitaの2017/12/19の記事です。

Chainerで学習したモデルをONNX-ChainerとNNVM/TVMを使ってAndroidへデプロイする

Usage

Chainerの学習済みモデルYOLOv2_tinyをONNXに変換

$ cd models/YOLOv2_tiny/
$ python export.py

macBookのCPU,GPUにデプロイする

$ python run_pc.py

AndroidのCPU,GPUにデプロイする

$ python -m tvm.exec.rpc_proxy 

Android側でRPCプロキシサーバーに接続後、以下のコマンドを実行

$ python run_mobile.py