python猫狗识别课程设计_码云社 | 砺锋科技-Python通过TensorFlow卷积神经网络实现猫狗识别 - 用代码改变世界...

#coding=utf-8

import tensorflow as tf

def inference(images, batch_size, n_classes):

with tf.variable_scope('conv1') as scope:

# 卷积盒的为 3*3 的卷积盒，图片厚度是3，输出是16个featuremap

weights = tf.get_variable('weights',

shape=[3, 3, 3, 16],

dtype=tf.float32,

initializer=tf.truncated_normal_initializer(stddev=0.1, dtype=tf.float32))

biases = tf.get_variable('biases',

shape=[16],

dtype=tf.float32,

initializer=tf.constant_initializer(0.1))

conv = tf.nn.conv2d(images, weights, strides=[1, 1, 1, 1], padding='SAME')

pre_activation = tf.nn.bias_add(conv, biases)

conv1 = tf.nn.relu(pre_activation, name=scope.name)

with tf.variable_scope('pooling1_lrn') as scope:

pool1 = tf.nn.max_pool(conv1, ksize=