DeepLab V3+ をCore MLに変換する

凍結グラフから変換します。

DeepLabをトレーニングする方法、凍結グラフとしてエクスポートする方法は下記の記事を参照していただければ。
DeepLabv3＋を自前のデータセットで学習させる

#手順

###１、必要なライブラリを用意
TensorFlow1をインストールしていない場合はインストール。

pip install tensorflow==1.x

Colabの場合。TensorFlow1を選択。

%tensorflow_version 1.x

Core ML Toolsをインストール。

pip install coremltools==4.0

###２、変換

import coremltools as ct

input = ct.ImageType(shape=(1, 513, 513, 3))
mlmodel = ct.convert('./frozen_graph.pb',inputs=[input])
mlmodel.save('./deeplabv3')

🐣

---

お仕事のご相談こちらまで
[e-mail]

Core MLを使ったアプリを作っています。
機械学習関連の情報を発信しています。

Twitter
Medium