self.avgpool = layers.GlobalAveragePooling2D()可以用来打平!
如果你的数据是3D的,要用 layers.GlobalAveragePooling1D()!效果:
TensorShape([8, 200, 256]) ===> TensorShape([8, 256])
# self.final_bn = layers.BatchNormalization()
# 接入Dense层之前,先打平!!!
self.avgpool = layers.GlobalAveragePooling2D()
self.fc1 = layers.Dense(num_classes*100)
self.fc2 = layers.Dense(1, activation=tf.nn.sigmoid)
也可以用layers.Flatten()
本文介绍如何利用GlobalAveragePooling2D层将3D数据转换为2D数据,以便于接入Dense全连接层。举例说明了从TensorShape([8,200,256])到TensorShape([8,256])的转换过程,并提到了另一种打平数据的方法-layers.Flatten()。

被折叠的 条评论
为什么被折叠?



