tf.nn.conv2d的参数

看到网上的很多讲解,给了很多例子,但是没有一个总结得比较符合自己的习惯的。所以自己总结一下:

tf.nn.conv2d(input, filter, strides, padding, use_cudnn_on_gpu=None, name=None)

input:输入的矩阵,要求是四维:[m, row_pixel, column_pixel, channel]
m即样本的数目,训练样本或者是测试样本的数目。
row_pixel就是图像的行像素数目
column_pixel就是图像的列像素数目
channel就是图像的通道数,简单来说就是,如果是黑白的图像,那通道数就是1,如果是彩色图像,那通道数就是3(因为RGB三个通道)

filter:卷积核,要求是四维:[row, column, input_matrix_channel, filter_channel]
row即卷积核的行数
column即卷积核的列数
input_matrix_channel即输入图像的通道数。如果后面的strides中设置了channel_strides不为1,那么这里就不再是输入图像的通道数,具体数值需要计算。不过一般都把后面strides中的channel_strides设置为1,所以没有例外的话,这样写也没啥太大的关系。
filter_channel即卷积核的通道数

strides:即步长,要求是一维的list:[m_strides, row_pixel_strides, column_pixel_strides, channel_strides]
m_strides即卷积核在输入矩阵的第一维(即样本数目)上进行卷积的步长
row_pixel_strides即卷积核在输入矩阵第二维(即行像素)上进行卷积的步长
column_pixel_strides即卷积核在输入矩阵第三维(即列像素)上进行卷积的步长
channel_strides即卷积核在输入矩阵第四维(即图像通道)上进行卷积的步长

其他的参数值的话,其他的帖子讲得都很清楚。这里不再赘述。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值