我想將“GlobalAveragePooling2D”和 Predication (Dense) 添加到我的基礎 ResNet50。如下圖
所以我這樣做了：
x=base_model.output
x = tf.keras.layers.GlobalAveragePooling2D()(x)
x = tf.keras.layers.Dense(2, activation='softmax')(x)
model = keras.models.Model(inputs=base_model.input, outputs=x)
model.summary()
但我得到了這個：
它們是不同的還是相同的，因為我認為我得到了不同的結果。
uj5u.com熱心網友回復：
它們是相同的，唯一的區別是您沒有為圖層命名。如果您這樣做，摘要將是相同的：
x = base_model.output
x = tf.keras.layers.GlobalAveragePooling2D(name = 'avg_pool')(x)
x = tf.keras.layers.Dense(2, activation='softmax', name = 'predictions')(x)
model = tf.keras.models.Model(inputs=base_model.input, outputs=x)
model.summary()
轉載請註明出處，本文鏈接：https://www.uj5u.com/qukuanlian/432309.html