質問編集履歴
1
fiveHundred さんに指摘していただいたところを修正しました。
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -10,23 +10,81 @@
|
|
|
10
10
|
|
|
11
11
|
|
|
12
12
|
|
|
13
|
-
|
|
13
|
+
|
|
14
14
|
|
|
15
15
|
エラーもなく変換できているのですが、変換されたtfliteファイルが1KBとサイズが小さすぎて困ってます
|
|
16
16
|
|
|
17
|
+
###変換をしたコード
|
|
18
|
+
|
|
17
19
|
```
|
|
20
|
+
|
|
21
|
+
#tensorflowをインストールする
|
|
22
|
+
|
|
23
|
+
!pip install tensorflow
|
|
24
|
+
|
|
25
|
+
#tensorflowのバージョンを確かめる
|
|
26
|
+
|
|
27
|
+
try:
|
|
28
|
+
|
|
29
|
+
%tensorflow_version 2.x
|
|
30
|
+
|
|
31
|
+
except Exception:
|
|
32
|
+
|
|
33
|
+
pass
|
|
18
34
|
|
|
19
35
|
|
|
20
36
|
|
|
37
|
+
import tensorflow as tf
|
|
38
|
+
|
|
39
|
+
from tensorflow.keras import layers
|
|
21
40
|
|
|
22
41
|
|
|
23
42
|
|
|
43
|
+
print("TF version:", tf.__version__)
|
|
44
|
+
|
|
45
|
+
#tfliteに変換する
|
|
46
|
+
|
|
47
|
+
# 基本的な関数を構築
|
|
48
|
+
|
|
49
|
+
root = tf.train.Checkpoint()
|
|
50
|
+
|
|
51
|
+
root.v1 = tf.Variable(3.)
|
|
52
|
+
|
|
53
|
+
root.v2 = tf.Variable(2.)
|
|
54
|
+
|
|
55
|
+
root.f = tf.function(lambda x: root.v1 * root.v2 * x)
|
|
56
|
+
|
|
57
|
+
|
|
58
|
+
|
|
59
|
+
# モデルを保存
|
|
60
|
+
|
|
61
|
+
export_dir = "/tmp/test_saved_model"
|
|
62
|
+
|
|
63
|
+
input_data = tf.constant(1., shape=[1, 1])
|
|
64
|
+
|
|
65
|
+
to_save = root.f.get_concrete_function(input_data)
|
|
66
|
+
|
|
67
|
+
tf.saved_model.save(root, export_dir, to_save)
|
|
68
|
+
|
|
69
|
+
|
|
70
|
+
|
|
71
|
+
# モデルを変換
|
|
72
|
+
|
|
73
|
+
converter = tf.lite.TFLiteConverter.from_saved_model(export_dir)
|
|
74
|
+
|
|
75
|
+
tflite_model = converter.convert()
|
|
76
|
+
|
|
77
|
+
!ls /tmp/test_saved_model
|
|
78
|
+
|
|
79
|
+
open('savedmodel-model.tflite', 'wb').write(tflite_model)
|
|
80
|
+
|
|
81
|
+
```
|
|
24
82
|
|
|
25
83
|
### 試したこと
|
|
26
84
|
|
|
27
85
|
学習済みデータを変えました。
|
|
28
86
|
|
|
29
|
-
p
|
|
87
|
+
pbファイルのほかにpbtxtファイルも付け足してやりました。
|
|
30
88
|
|
|
31
89
|
### 補足情報
|
|
32
90
|
|