网上找了挺久没有能直接拿来用的,索性自己搭一个吧,精简结构图:
下面上代码:
# alexnet
# 构建模型
def inference(input_tensor, train, regularizer):
with tf.variable_scope('layer1-conv1'):#通过tf.variable_scope函数控制tf.get_variable函数来获取以及创建过的变量
conv1_weights = tf.get_variable("weight",[11,11,3,96],initializer=tf.truncated_normal_initializer(stddev=0.1))
conv1_biases = tf.get_variable("bias", [96], initializer=tf.constant_initializer(0.0))
conv1 = tf.nn.conv2d(input_tensor, conv1_weights, strides=[1, 4, 4, 1], padding='SAME')#strides=【1,1,1,1】表示滑动步长为1
relu1 = tf.nn.relu(tf.nn.bias_add(conv1, conv1_biases))
relu1=tf.nn.local_response_normalization(relu1, depth_radius=2, alpha=1e-4, beta=0.75, name='norm1', bias=1)
with t