前提・実現したいこと
tensorflowの学習済みデータ(saved_model)を下のURLのサイトを参考にしてgooglecolabでtflite形式に変換したいと思っています
サイトのURL:https://rightcode.co.jp/blog/information-technology/tensorflow-model-file-conversion-important-point
発生している問題・エラーメッセージ
エラーもなく変換できているのですが、変換されたtfliteファイルが1KBとサイズが小さすぎて困ってます
###変換をしたコード
#tensorflowをインストールする !pip install tensorflow #tensorflowのバージョンを確かめる try: %tensorflow_version 2.x except Exception: pass import tensorflow as tf from tensorflow.keras import layers print("TF version:", tf.__version__) #tfliteに変換する # 基本的な関数を構築 root = tf.train.Checkpoint() root.v1 = tf.Variable(3.) root.v2 = tf.Variable(2.) root.f = tf.function(lambda x: root.v1 * root.v2 * x) # モデルを保存 export_dir = "/tmp/test_saved_model" input_data = tf.constant(1., shape=[1, 1]) to_save = root.f.get_concrete_function(input_data) tf.saved_model.save(root, export_dir, to_save) # モデルを変換 converter = tf.lite.TFLiteConverter.from_saved_model(export_dir) tflite_model = converter.convert() !ls /tmp/test_saved_model open('savedmodel-model.tflite', 'wb').write(tflite_model)
試したこと
学習済みデータを変えました。
pbファイルのほかにpbtxtファイルも付け足してやりました。
補足情報
学習済みデータは、下のURLのサイトを参考にしてやりました。
サイトのURL:https://qiita.com/tsota/items/123514cbfd036e6bd808#1-cloud-annotations%E3%81%A7%E3%82%A2%E3%83%8E%E3%83%86%E3%83%BC%E3%82%B7%E3%83%A7%E3%83%B3
誰か教えてくれると嬉しいです。
回答1件
あなたの回答
tips
プレビュー