TFCoreMLで変換したモデルの出力がおかしい時の対処法

WWDC2020あたりを境に、tfcoremlで変換したGANの出力が全部真っ白になりました。

対処法

Core ML Tools 4.0をインストールして変換し直す。

pip install coremltools==4.0b2

tfcoremlに比べて、変換メソッドのオプションがシンプルに変わっています。

import coremltools as ct
image_input = ct.ImageType(shape=(1, 256, 256, 3,),
                             scale=2/255,bias=[-1,-1,-1])

coreml_model = ct.convert(
        yourmodel,
        inputs=[image_input],
        )
coreml_model.save("yourmodel.mlmodel")

これで出力されました。

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

Twitter
MLBoysチャンネル
Medium

相棒
note