# -*- coding:utf-8 -*-
#
# AlexNet model
import tensorflow as tf
import numpy as np
FLAGS = tf.app.flags.FLAGS
tf.app.flags.DEFINE_integer('num_class', 10, "number of classes.")
#显示网络每一层结构的函数
def print_layer(t):
print(t.op.name, ' ', t.get_shape().as_list())
def alexnet(images):
# conv1
with tf.name_scope('conv1') as scope:
kernel = tf.Variable(tf.truncated_normal([11,11,3,64], stddev=0.1), name='weights')
biases = tf.Variable(tf.constant(0.0, shape=[64], trainable=True, name='biases')
conv_plus_b = tf.nn.bias_add(tf.nn.conv2d(images, kernel, [1,4,4,1], padding='SAME'), biases)
conv1 = tf.nn.relu(conv_plus_b, name=scope)
print_layer(conv1)
# LRN层, pool层
lrn1 = tf.nn.lrn(