tensorflow学习:池化层(pooling)和全连接层(dense)

15 篇文章 0 订阅
10 篇文章 2 订阅

一、池化层的github中的比较官方的定义:
池化层定义
有最大值池化和均值池化

  1. tf.layers.max_pooling2d
max_pooling2d(
    inputs,
    pool_size,
    strides,
    padding='valid',
    data_format='channels_last',
    name=None
)
  • inputs: 进行池化的数据。
  • pool_size: 池化的核大小(pool_height, pool_width),如[3,3].如果长宽相等,也可以直接设置为一个数,如pool_size=3.
  • strides: 池化的滑动步长。可以设置为[1,1]这样的两个整数.也可以直接设置为一个数,如strides=2
  • padding: 边缘填充,’same’ 和’valid‘选其一。默认为valid
  • data_format: 输入数据格式,默认为channels_last ,即 (batch, height, width, channels),也可以设置为channels_first 对应 (batch, channels, height, width).
  • name: 层的名字。
    例如:
pool1=tf.layers.max_pooling2d(inputs=x, pool_size=[2, 2], strides=2)

池化层一般放在卷积之后的位置上,如:

conv=tf.layers.conv2d(
      inputs=x,
      filters=32,
      kernel_size=[5, 5],
      padding="same",
      activation=tf.nn.relu)
pool=tf.layers.max_pooling2d(inputs=conv, pool_size=[2, 2], strides=2)

2、tf.layers.average_pooling2d

average_pooling2d(
    inputs,
    pool_size,
    strides,
    padding='valid',
    data_format='channels_last',
    name=None
)

参数的具体含义跟最大值池化的含义一样,这里就不再进行过多的陈述。

二、全连接dense层定义在github中也有相应的定义:
全连接层的定义

  1. tf.layers.dense
dense(
    inputs,
    units,
    activation=None,
    use_bias=True,
    kernel_initializer=None,
    bias_initializer=tf.zeros_initializer(),
    kernel_regularizer=None,
    bias_regularizer=None,
    activity_regularizer=None,
    trainable=True,
    name=None,
    reuse=None
)
  • inputs: 输入数据,2维tensor.
  • units: 该层的神经单元结点数。
  • activation: 激活函数.
  • use_bias: Boolean型,是否使用偏置项.
  • kernel_initializer: 卷积核的初始化器.
  • bias_initializer: 偏置项的初始化器,默认初始化为0.
  • kernel_regularizer: 卷积核化的正则化,可选.
  • bias_regularizer: 偏置项的正则化,可选.
  • activity_regularizer: 输出的正则化函数.
  • trainable: Boolean型,表明该层的参数是否参与训练。如果为真则变量加入到图集合中
  • GraphKeys.TRAINABLE_VARIABLES (see tf.Variable).
  • name: 层的名字.
  • reuse: Boolean型, 是否重复使用参数.
    全连接层执行操作 outputs = activation(inputs.kernel + bias)
    如果执行结果不想进行激活操作,则设置activation=None。
    例如:
#全连接层
dense1 = tf.layers.dense(inputs=pool3, units=1024, activation=tf.nn.relu)
dense2= tf.layers.dense(inputs=dense1, units=512, activation=tf.nn.relu)
logits= tf.layers.dense(inputs=dense2, units=10, activation=None)

最后可以对全连接层进行正则化约束:

dense1 = tf.layers.dense(inputs=pool3, units=1024, activation=tf.nn.relu,kernel_regularizer=tf.contrib.layers.l2_regularizer(0.003))
  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
dense全连接层是指在神经网络中的一种次结构。在TensorFlow中,可以使用tf.layers.dense函数来创建全连接层全连接层有两个主要的作用:一是将前一的输出变为一维向量,这个过程称为Flatten;二是将这个一维向量与权重相乘并加上偏置,然后通过激活函数进行非线性变换,得到最终的输出结果。全连接层通常用于将卷积的输出转换成与任务相关的特征表示,最后用于分类、回归或其他预测任务。在一个典型的神经网络中,通常会有多个全连接层,每个全连接层的输出会作为下一输入。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *3* [tensorflow 学习笔记(八)- 池化层pooling)和全连接层(dense)](https://blog.csdn.net/m0_37167788/article/details/79072681)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *2* [全连接层、卷积](https://blog.csdn.net/weixin_41624572/article/details/103006032)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值