tf.contrib.slim.conv2d ()和tf.nn.conv2d()的使用区别

在使用tensorflow时,我们常见的卷积操作是tf.contrib.slim.conv2d ()和tf.nn.conv2d(),这里简略给出用法及区别。
1.tf.contrib.slim.conv2d ()

tf.contrib.slim.conv2d (inputs,
            num_outputs,#[卷积核个数]
            kernel_size,#[高度,宽度]
            stride=1,#步长
            padding='SAME',#VALID

)

2.tf.nn.conv2d()

tf.nn.conv2d(inputs,
        filter,([卷积核的高度,卷积核的宽度,图像通道数,卷积核个数])
        strides,
        padding,
)

猜你喜欢

转载自blog.csdn.net/xuan_zizizi/article/details/78518117