卷积神经网络图像识别catvsdog代码

卷积神经网络图像识别catvsdog代码如果开发者对于本文件有需要的可以参考。

with tf.variable_scope('conv1'): # 第一层卷积 输入图片数据(?,IMGSIZE,IMGSIZE,3)
W1 = tf.Variable(tf.random_normal([3, 3, 3, 16])) # 卷积核3*3，输入通道3，输出通道16
L1 = tf.nn.conv2d(X, W1, strides=[1, 1, 1, 1], padding='SAME') # 卷积输出 (?,IMGSIZE,IMGSIZE,16)
L1 = tf.nn.relu(L1)
L1 = tf.nn.max_pool(L1, ksize=[1, 2, 2, 1], strides=[1, 2, 2, 1], padding='SAME') # 池化输出(?,IMGSIZE/2,IMGSIZE/2,16)