Couche de regroupement moyenne 2D
keras.layers.AveragePooling2D(pool_size=(2, 2),
strides=None,
padding='valid',
data_format=None)
Paramètres détaillés
- pool_size : taille de la fenêtre de regroupement
- foulées : la foulée de regroupement, la valeur par défaut est égale à
pool_size
- padding : '
VALID
' ou ' SAME
', ' VALID
' signifie pas de remplissage, ' SAME
' signifie remplissage avec 0
- data_format : Indique l'ordre des dimensions du tenseur d'entrée, la valeur par défaut est [batch, height, width, channel]
exemple
from tensorflow.keras.layers import AveragePooling2D
import tensorflow as tf
import numpy as np
# 定义一个平均池化层,用0填充
pool1 = AveragePooling2D(pool_size=(2, 2),
strides=None,
padding='SAME',
data_format=None)
# 定义一个平均池化层,不填充
pool2 = AveragePooling2D(pool_size=(2, 2),
strides=None,
padding='VALID',
data_format=None)
# 生成一个维度为[64, 101, 101, 3]的矩阵
x = np.random.random((64, 101, 101, 3))
# 转成tensor类型,第一个维度64表示batch
# numpy中的数据类型和tensorflow中的数据类型完全兼容,所以这一步可以省略
x = tf.convert_to_tensor(x)
print(x.shape) # [64, 101, 101, 3]
# 进行平均池化
y1 = pool1(x)
print(y1.shape) # [64, 51, 51, 3]
# 进行平均池化
y2 = pool2(x)
print(y2.shape) # [64, 50, 50, 3]