質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.35%

Q&A

解決済

1回答

1398閲覧

googlecolabでtflite形式に変換したい

children

総合スコア89

0グッド

0クリップ

投稿2020/05/08 08:48

編集2020/05/08 22:47

前提・実現したいこと

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
誰か教えてくれると嬉しいです。

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

fiveHundred

2020/05/08 12:09

「pdファイル」ではなく「pbファイル」なのでは? また、サイトだけ提示しても、実際の行った手順が分からないので、実際に学習や変換を行ったコードなども質問文に記載したほうがよろしいかと思います。
children

2020/05/08 22:50 編集

修正の依頼ありがとうございます。依頼されたところを修正ました。
guest

回答1

0

ベストアンサー

提示のコードのモデルは大きくないので、1KBに収まってもなんら不思議ではありません。
とりあえず、tfliteの中身をチェック(Netronがおすすめ)してみてください。


あと、質問とは関係ないですが、

python

1#tensorflowをインストールする 2!pip install tensorflow 3#tensorflowのバージョンを確かめる 4try: 5 %tensorflow_version 2.x 6except Exception: 7 pass

colabでは、初めからtensorflowはインストールされており、%tensorflow_version 2.xはそのtensorflowのバージョンを切り替えるためのものです。
従って、!pip install tensorflowは不要ですし、バージョンを表示させているのは、%tensorflow_version 2.xではなく以下の箇所になります。

Python

1print("TF version:", tf.__version__)

投稿2020/05/09 01:38

fiveHundred

総合スコア10152

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

fiveHundred

2020/05/13 06:51

Piを持ってないので、おそらく私には分かりませんが、他の人が分かるようにエラーの内容を記載するほうがよろしいかと思います。
guest

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.35%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問