Tensorflow卷积API参数说明
tf.nn.conv2d(
input,
filter,
strides,
padding,
use_cudnn_on_gpu=True,
data_format='NHWC',
dilations=[1, 1, 1, 1],
name=None
)
-
input : 代表原图,是一个数组里面包括[batch_size, in_height, in_width, n_channels]
- batch_size :每次输入的批次,如果每次输入一个图片,那就是1.
- in_height :图片矩阵的高度。
- in_width :图片矩阵的宽带。
- n_channels : 图片矩阵的通道数,例如彩色图片那通道数就是3了。
-
filter : 代表卷积核,也是一个数组包括[filter_height, filter_width, in_channels, out_channels]
- filter_height :卷积核的高度。
- filter_width:卷积核的