KerasのモデルをtfLiteに変換できない。

実現したいこと

GoogleColab上で、KerasのモデルをTFLiteに変換して保存したい。

発生している問題・分からないこと

　1週間ほど前まで動いていた、画像分類モデルを作るコードが突然動かなくなりました。モデル自体は作れるが、これをTFLiteのモデルとして保存することができません。エラーメッセージを見ると、「model.save('/content/saved_model/myModel')」という部分で、myModelに拡張子をつけてmyModel.kerasとせよとあったので、そうすると、この部分は動き、Kerasのモデルを保存することができました。
ところが、そのあとでこれをTFLiteに保存する部分で、エラーが起きます。変換すべきモデルが指定されたパスに存在しないといったエラーメッセージが出るが、実際にはそこにはモデル（myModel.keras）が表示されています。ただ、以前はここにはフォルダがあって、その中に.pbという拡張子のついたファイルなどいくつかがあったように記憶していますので、model.save()のところで拡張子をつけたことによって、保存のしかたが変わってしまったような気がしています。

エラーメッセージ

error
1OSError                                   Traceback (most recent call last)
2<ipython-input-30-906685e7ef33> in <cell line: 7>()
3      5 model.save('/content/saved_model/myModel.keras')
4      6 
5----> 7 converter = tf.lite.TFLiteConverter.from_saved_model('/content/saved_model/myModel.keras')
6      8 tflite_model = converter.convert()
7      9 
8
94 frames
10/usr/local/lib/python3.10/dist-packages/tensorflow/python/saved_model/loader_impl.py in parse_saved_model(export_dir)
11    117       raise IOError(f"Cannot parse file {path_to_pbtxt}: {str(e)}.") from e
12    118   else:
13--> 119     raise IOError(
14    120         f"SavedModel file does not exist at: {export_dir}{os.path.sep}"
15    121         f"{{{constants.SAVED_MODEL_FILENAME_PBTXT}|"
16
17OSError: SavedModel file does not exist at: /content/saved_model/myModel.keras/{saved_model.pbtxt|saved_model.pb}

該当のソースコード

Python
1model.save('/content/saved_model/myModel.keras')
2converter = tf.lite.TFLiteConverter.from_saved_model('/content/saved_model/myModel.keras')
3tflite_model = converter.convert()
4

試したこと・調べたこと

teratailやGoogle等で検索した
ソースコードを自分なりに変更した
知人に聞いた
その他

上記の詳細・結果

モデルを保存する方法として、.kerasという拡張子をつける以外に.h5という拡張子をつける方法（レガシー）もあるということなので、それを試したが結果は変わらなかった。
Googleで類似の問題について調べたが、該当するものを見つけられなかった。

補足

特になし

行動規範の内容に同意します

回答1件

自己解決

　投稿の後で調べたところ、 tf.lite.TFLiteConverter.from_saved_model()の引数とすべきsaved_modelというモデルの形式が使えなくなっていること、そのために拡張子（.keras）をつけて保存するようにとのエラーメッセージが出たことがわかりました。さらに、.kerasの形式のモデルはsaved_modelと同じ方法ではTFLiteに変換できず、別のメソッドを使う必要があることと、実際に変換する実際の方法が次のサイトからわかりました。
https://www.tensorflow.org/lite/models/convert/convert_models?hl=ja

　これによって、TFLiteのモデルを作れたのですが、これにはメタデータがついていませんので、TFLiteのモデルにメタデータをつけるために今まで使っていたコードを適用してみると、それが動かないという状況が発生しています。従って、問題は解決していないのですが、これはまた別のテーマと思われるので、少しやってみて、うまくいかなければ再び質問させていただこうと思っています。

　→その後、いろいろやってみてとりあえず解決しました。というのは、メタデータの追加は次のサイトにある方法で行ったのですが、
https://www.tensorflow.org/lite/models/convert/metadata?hl=ja
その際、モデルを保存するフォルダ名を "test_model" としていたのですが、これを "model_test" に書き換えるとうまくメタデータをつけることができました。どうも、フォルダ名が小文字の"t"で始まると、VSC上でその部分の色が微妙に変わり（もっとはっきり変えてくれたらいいのですが）、何らかのエラーが起きているようです。色が変わる文字は"t"の他にも"a"、"b"など多くあります。全部調べていませんが"c"、"d"、"e"などは変わりません。また、大文字にしておくと色は変わりません。
ということで、問題自体は解決したのですが、この現象はとても奇妙に思えます。どなたかご存じの方がおられたら、教えていただけると幸いです。

投稿2024/09/06 22:19

編集2024/09/06 23:41

masa-nakamura

総合スコア5