TensorFlow实战系列12--池化层网络结构

最新推荐文章于 2021-04-18 12:49:26 发布

缘定三石

最新推荐文章于 2021-04-18 12:49:26 发布

阅读量1.4k

点赞数

分类专栏： TensorFlow

本文链接：https://blog.csdn.net/tian_qing_lei/article/details/79270532

版权

本文介绍了卷积神经网络中的池化层，包括最大池化层和平均池化层的作用，以及如何在TensorFlow中实现这两种池化层的前向传播算法。池化层通过减小矩阵尺寸，加速计算，防止过拟合。最大池化层通常使用`tf.nn.max_pool`函数，平均池化层则使用`tf.nn.avg_pool`函数，步长和填充方式与卷积层类似。

摘要由CSDN通过智能技术生成

在卷积神经网络中,卷积层之间往往会加上一个池化层(poolinglayer)。池化层可以非常有效地缩小矩阵的尺寸,从而减少最后全连接层中的参数。使用池化层既可以加快计算速度也有防止过拟合问题的作用。和卷积层类似,池化层前向传播的过程也是通过移动一个类似过滤器的结构完成的。不过池化层过滤器中的计算不是节点的加权和,而是采用更加简单的最大值或者平均值运算。使用最大值操作的池化层被称之为最大池化层(max pooling),这是被使用得最多的池化层结构。使用平均值操作的池化层被称之为平均池化层(average pooling)。

与卷积层的过滤器类似,池化层的过滤器也需要人工设定过滤器的尺寸、是否使用全 0 填充以及过滤器移动的步长等设置,而且这些设置的意义也是一样的。卷积层和池化层中过滤器移动的方式是相似的,唯一的区别在于卷积层使用的过滤器是横跨整个深度的,而池化层使用的过滤器只影响一个深度上的节点。所以池化层的过滤器除了在长和宽两个维度移动之外,它还需要在深度这个维度移动。下面的 TensorFlow 程序实现了最大池化层的前向传播算法。

# tf.nn. max_pool实现了最大池化层的前向传播过程,它的参数和tf.nn.conv2d函数类似。
# ksize提供了过滤器的尺寸,strides提供了步长信息,padding提供了是否使用全0填充。
pool = tf.nn.max_pool(actived_conv, ksize=[1, 3, 3, 1],