TensorFlow1のモデルをCore MLに変換する

Last updated at 2020-09-29Posted at 2020-09-29

#グラフからの変換

TensorFlow1からの変換は、
１、凍結グラフからの変換
２、グラフからの直接変換
があります。
１、の凍結グラフの作成方法については「AnimeGANv2をCore MLに変換してiOSでつかう」とおなじです。

#TensorFlow1の変換

###１、凍結グラフからの変換


mlmodel = ct.convert('inferrence.pb')

###２、グラフからの直接変換


mlmodel = ct.convert(self.sess.graph)

グラフからの直接変換は、sess.runしているスクリプトの下に書き込んでしまうのがおすすめです。

お仕事のご相談こちらまで
rockyshikoku@gmail.com

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