Tensorflow中2D卷积API使用

TF中用于2D卷积操作的api是tf.nn.conv_2d，源码位置位于tensorflow/python/ops/gen_nn_ops.py

接口定义为：

　　conv2d(input, filter, strides,padding, use_cudnn_on_gpu=True, data_format=‘NHWC’，name='None')

其中

　　input的shape是[batch, in_height, in_width, in_channels]　　

　　filter的shape是[filter_height, filter_width, in_channels, out_channels]

　　strides在的通常要写成[1, stride, stride, 1]，也就是说第0个参数和第三个参数都是固定1，中间分别是x方向和y方向的步长，通常是一样的。

　　padding是一个字符串类型，有SAME和VALID两个取值。

返回的同样是一个tensor。

再说一下tf.nn.bias.add这个api

接口定义是：

bias_add(value, bias, data_format=None, name = None)

value是一个tensor，可以是float的，double，int32等等类型的，value可以是任意维度的tensor，但是value的最后一维的size要和bias相同，类型可以不一样，可以转化就可以例如int转double等。

例如在下面的代码中经过卷积操作之后，输入是28*28*1的单通道图像，输出编程了24*24*32的特征图，偏置也是32维的tensor，特征图i对应+bias[i]，输出还是24*24*32。同一个特征图共享了卷积核的5*5个参数和1个bias参数，也就是说这里一共有（5*5+1）*32个参数。

#从mnist数据集中读数据，定义一个占位符先
x = tf.placeholder(tf.float32, [None, 28*28])
#将x reshape成一个4D的tensor， 分别是batch size， height， width，channel
x = x.reshape(x, shape=[-1, 28,28,1])
#这个变量是卷积核，卷积核大小是5*5，通道数1，卷积核格式32个
weight = tf.Variable(tf.random_normal([5,5,1,32]))
bias = tf.Variable(tf.random_normal([32]))
stride = 1
x = tf.nn.conv2d(x, weight, strides = [1, stride, stride, 1], padding= 'SAME')
#再加上偏置量
x = tf.nn.bias_add(x, bias)

　通过上面的代码应该就可以很清楚的明白tensorflow中是如何做2维卷积的了。

另外一个用于实现2D卷积的API是tf.layers.conv2d，实际上和上面的这个API并没有太大区别，只是参数更多一些，使用的时候二者都可以。

后续再说一说CNN中的转置卷积（deconvolutional network）在tf中的api的使用，以及计算方式。

转载于:https://www.cnblogs.com/hellozlh/p/8251850.html

Tensorflow中2D卷积API使用相关推荐

【Tensorflow】Tensorflow中的卷积函数（conv2d、slim.conv2d、depthwise_conv2d、conv2d_transpose）
[fishing-pan:https://blog.csdn.net/u013921430 转载请注明出处] 前言卷积是卷积神经网络中最主要.最重要的运算.想必大家最早接触卷积的概念就是在初高中 ...
如何使用TensorFlow中的Dataset API
翻译 | AI科技大本营参与 | zzq 审校 | reason_W 本文已更新至TensorFlow1.5版本我们知道,在TensorFlow中可以使用feed-dict的方式输入数据信息,但是 ...
TensorFlow中padding卷积的两种方式“SAME”和“VALID”
最近在用tensorflow搭建卷积神经网络,遇到了一个比较棘手的问题,我一直理解的padding有两个值,一个是SAME,一个是VALID,如果padding设置为SAME,则说明输入图片大小和输出 ...
通过图+代码来理解tensorflow中反卷积
反卷积这个东西老是容易忘,而且很多文章理论讲的很详细,但反卷积实际怎么操作的却没有概念,因此想以自己喜欢的方式(直接上图和代码)写一篇,以便随时翻阅. 卷积 tf中的padding方式有两种,SAME ...
tensorflow中转置卷积运算例子
import tensorflow as tf import numpy as np import tensorflow as keras from tensorflow.keras import l ...
TensorFlow 中的 Conv2DOp
TensorFlow 中的2D 卷积主要依赖外部库,如 cuDNN.cuBLAS.ROCm 和 hfp/libxsmm,仅 DeepConv2D<CPUDevice, T> 为源码实现. ...
Tensorflow中部分API弃用(tf.sub/tf.mul/tf.neg)
Tensorflow中废弃的API及替代 tf.mul , tf.sub , tf.neg 已经废弃分别可用tf.multiply , tf.subtract , tf.negative ...
CNN中的卷积核及TensorFlow中卷积的各种实现
声明: 1. 我和每一个应该看这篇博文的人一样,都是初学者,都是小菜鸟,我发布博文只是希望加深学习印象并与大家讨论. 2. 我不确定的地方用了"应该"二字首先,通俗说一下,CNN ...
Tensorflow中卷积的padding操作
上海站 | 高性能计算之GPU CUDA培训 4月13-15日三天密集式学习快速带你晋级阅读全文 > 正文共975字,2张图,预计阅读时间3分钟. 之前一直对tensorflow的pad ...

Tensorflow中2D卷积API使用

Tensorflow中2D卷积API使用相关推荐

最新文章

热门文章