本文整理汇总了Python中tensorflow.python.keras.layers.GlobalAveragePooling2D方法的典型用法代码示例。如果您正苦于以下问题：Python layers.GlobalAveragePooling2D方法的具体用法？Python layers.GlobalAveragePooling2D怎么用？Python layers.GlobalAveragePooling2D使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类tensorflow.python.keras.layers的用法示例。
在下文中一共展示了layers.GlobalAveragePooling2D方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: _define_model
# 需要导入模块: from tensorflow.python.keras import layers [as 别名]
# 或者: from tensorflow.python.keras.layers import GlobalAveragePooling2D [as 别名]
def _define_model(output_layer=-1):
'''Define a pre-trained MobileNet model.
Args:
output_layer: the number of layer that output.
Returns:
Class of keras model with weights.
'''
base_model = MobileNet(weights='imagenet', include_top=False, input_shape=(224, 224, 3))
output = base_model.layers[output_layer].output
output = GlobalAveragePooling2D()(output)
model = Model(inputs=base_model.input, outputs=output)
return model