質問編集履歴
1
コードを修正し動作確認を行いました。
test
CHANGED
|
File without changes
|
test
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
### 前提・実現したいこと
|
|
2
2
|
|
|
3
|
-
Kerasで
|
|
3
|
+
Kerasで作成したモデル（学習済みEfficientNetB0 + 全結合層）について以下の２つを実現したいです。
|
|
4
4
|
|
|
5
5
|
1. **model.summary()で全結合層の詳細を表示したい**
|
|
6
6
|
|
|
@@ -18,17 +18,23 @@
|
|
|
18
18
|
|
|
19
19
|
###モデル
|
|
20
20
|
|
|
21
|
-
モデルは以下のように作成し
|
|
21
|
+
モデルは以下のように作成しました。
|
|
22
|
-
|
|
22
|
+
|
|
23
|
-
```
|
|
23
|
+
```
|
|
24
|
-
|
|
24
|
+
|
|
25
|
-
from keras.models import Sequential, Model
|
|
25
|
+
from keras.models import Sequential, Model, load_model
|
|
26
26
|
|
|
27
27
|
from keras.layers import Flatten, Dense,Input, Dropout
|
|
28
28
|
|
|
29
29
|
from efficientnet.keras import EfficientNetB0
|
|
30
30
|
|
|
31
|
-
|
|
31
|
+
from keras import optimizers
|
|
32
|
+
|
|
33
|
+
|
|
34
|
+
|
|
35
|
+
|
|
36
|
+
|
|
37
|
+
input_shape = (9, 128, 3)
|
|
32
38
|
|
|
33
39
|
input_tensor = Input(shape=input_shape)
|
|
34
40
|
|
|
@@ -52,12 +58,18 @@
|
|
|
52
58
|
|
|
53
59
|
|
|
54
60
|
|
|
55
|
-
top_model = Model(input=main_model.input, output=top_model(main_model.output))
|
|
61
|
+
top_model = Model(inputs=main_model.input, outputs=top_model(main_model.output))
|
|
56
62
|
|
|
57
63
|
top_model.compile(loss='categorical_crossentropy',optimizer=optimizers.SGD(lr=1e-3, momentum=0.9),metrics=['accuracy'])
|
|
58
64
|
|
|
59
65
|
|
|
60
66
|
|
|
67
|
+
path_model = "./models/EfficientNetB0_FC_not_work.h5"
|
|
68
|
+
|
|
69
|
+
top_model.save(path_model)
|
|
70
|
+
|
|
71
|
+
|
|
72
|
+
|
|
61
73
|
```
|
|
62
74
|
|
|
63
75
|
|
|
@@ -68,13 +80,13 @@
|
|
|
68
80
|
|
|
69
81
|
|
|
70
82
|
|
|
71
|
-
上で作成
|
|
83
|
+
上で作成したモデルを読み込みmodel.summary()を表示したところ、以下のように全結合層はひとまとめにSequentialとなりました。
|
|
72
|
-
|
|
84
|
+
|
|
73
|
-
```
|
|
85
|
+
```
|
|
74
|
-
|
|
86
|
+
|
|
75
|
-
|
|
87
|
+
path_model = "./models/EfficientNetB0_FC_not_work.h5"
|
|
76
|
-
|
|
88
|
+
|
|
77
|
-
model =
|
|
89
|
+
model = load_model(path_model, compile=False)
|
|
78
90
|
|
|
79
91
|
print(model.summary())
|
|
80
92
|
|
|
@@ -142,6 +154,10 @@
|
|
|
142
154
|
|
|
143
155
|
Non-trainable params: 42,016
|
|
144
156
|
|
|
157
|
+
__________________________________________________________________________________________________
|
|
158
|
+
|
|
159
|
+
None
|
|
160
|
+
|
|
145
161
|
```
|
|
146
162
|
|
|
147
163
|
|
|
@@ -210,14 +226,16 @@
|
|
|
210
226
|
|
|
211
227
|
top_activation (Activation) (None, 1, 4, 1280) 0 top_bn[0][0]
|
|
212
228
|
|
|
213
|
-
==================================================================================================
|
|
214
|
-
|
|
215
229
|
Total params: 4,049,564
|
|
216
230
|
|
|
217
231
|
Trainable params: 4,007,548
|
|
218
232
|
|
|
219
233
|
Non-trainable params: 42,016
|
|
220
234
|
|
|
235
|
+
__________________________________________________________________________________________________
|
|
236
|
+
|
|
237
|
+
None
|
|
238
|
+
|
|
221
239
|
```
|
|
222
240
|
|
|
223
241
|
|