M1 MacでTensorflowLiteによるKerasモデルの圧縮

  • 2022.09.30
  • IT
NO IMAGE

今回は、h5形式で保存したモデルをtfliteで保存する方法を紹介します。

このQiitaのTensorflowLiteによるKerasモデルの圧縮  という記事を参考にしました。

私の環境ではライブラリが更新されて動かなかったので、動くようにしました。

利用環境

利用環境は、M1 Macです。

Pythonのライブラリは下記です。不要なものがたくさん入っていますが、Tensorflow周りのバージョンは参考になると思います。

 

 

コード

下記がコードです。難しいことも特にないですが…。

補足

読み込むモデルは、

model.save()で保存したものであり、model.save_weights()で保存したものではNGです。

オプションなどは公式ドキュメントが参考になります。

 

以上です。