
PyTorch MobileのiOSサンプルを読む
coremltools 4.0でPyTorchモデルを直接Core MLモデルに変換できるようになったのでiOS向けとしてはもうそれで良い気もするが、もしかしたら仕事でPyTorch Mobileを使う必要が出てくるかもしれないので、サンプルを試しつつ中身を見つつで感触を得ておく。
https://pytorch.org/mobile/ios/
## サンプルを試す
公式のサンプルリポジトリがあり、
HelloWorldというサンプルがある。
Podfileがあるのでpod installしてHelloWorld.xcworkspaceを開きビルド。
こんな感じのシンプルな画像分類サンプル。
## サンプルの中身を見る
サンプルの中身を見ると、TorchModule.h/mmというブリッジ用のObjective-C++コードがあり、位置づけとしてはTensorFlowのモデルを(Core MLモデルに変換せず)iOSで用いる際のC++コード群と同じっぽい。CocoaPodsで簡単にインストールできるようにまとめてくれているというだけで。
最後まで読んでいただきありがとうございます!もし参考になる部分があれば、スキを押していただけると励みになります。 Twitterもフォローしていただけたら嬉しいです。 https://twitter.com/shu223/