バイセル Tech Blog

バイセル Tech Blogは株式会社BuySellTechnologiesのエンジニア達が知見・発見を共有する技術ブログです。

Pytorch(skorch)で学習したモデルを使い、iOSで画像分類をする [1/3]

テクノロジー開発部の村上です。現在はアーキテクチャ周りを担当しています。
弊社で社内向けにお酒判定iOSアプリを作成したので、そこで使った技術を3回に渡って紹介したいと思います。

  1. Skorch (Pytorchを使ったライブラリ) でCNNモデル作成
  2. Pytorchモデルを -> ONNX -> CoreMLモデル とiOSで使える形に変換
  3. CoreMLモデルをSwiftのPlaygroundで検証してみる
続きを読む