TensorFlowで学習させたモデルをTensorFlowLiteに変換したい

Question

リンク内容

こちらのサイトを参考に記事内の

リンク内容

こちらのソースコードを利用し、機械学習を行なっていました

saved_model_dir = base_dir+'保存したい場所'
tf.saved_model.save(model, saved_model_dir)

保存したい場所をFlowerとしたら同じGoogleドライブ内にimagesFlowerというフォルダが作成されたところまでは問題なかったのですが次のTensorFlowLiteに変換するためのコードを入力したところ

converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
tflite_model = converter.convert()

with open(base_dir+'モデルの名前'.tflite', 'wb') as f:
  f.write(tflite_model)

モデルの名前はFlowerにしています

このようなエラーが出てしまいました

  File "<ipython-input-21-a61b39a797df>", line 4
    with open(base_dir+'Flower'.tflite', 'wb') as f:
                                         ^
SyntaxError: invalid syntax

どなたか解決方法がわかる方いらっしゃいましたら教えていただけると幸いです

追記

converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
tflite_model = converter.convert()

with open(base_dir+'モデルの名前'.tflite', 'wb') as f:
  f.write(tflite_model)

ここの部分のモデルの名前という部分は気にならなかったのですが、.tfliteの後ろとwb前後の'が気になり外して下記のように実行したところ

converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)
tflite_model = converter.convert()

with open(base_dir+'Flower'.tflite, wb) as f:
  f.write(tflite_model)

以下のエラーが出てしまいました

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-31-70bc62405c67> in <module>()
      2 tflite_model = converter.convert()
      3 
----> 4 with open(base_dir+'Flower'.tflite, wb) as f:
      5   f.write(tflite_model)

AttributeError: 'str' object has no attribute 'tflite'

Answer 1

参考にしているGoogle Colab上のコードがそもそも間違っていますね。シングルクォーテーションが余分にあります。

質問欄にある修正後のコードですが、.tfliteはファイルの拡張子なわけですから

with open(base_dir+'Flower.tflite', 'wb') as f:

として.tfliteも文字列に含める必要があります。それとwbも文字列にしてください。

Comments

[baruterute]

ありがとうございます、余分にあったシングルクォーテーションを消して.tfliteを含めたところ無事変換することができました

ありがとうございました。