TensorFlow网络构建模板

# 2D网络
import tensorflow as tf

initial_value = tf.truncated_normal([filter_size, filter_size, in_channels, out_channels], 0.0, 0.001)
filters = tf.Variable(initial_value, name=var_name)
initial_value = tf.truncated_normal([out_channels], .0, .001)
biases = tf.Variable(initial_value, name=var_name)
conv = tf.nn.conv2d(bottom, filt, [1, 1, 1, 1], padding='SAME')
bias = tf.nn.bias_add(conv, conv_biases)
relu = tf.nn.relu(bias)

# 3D网络

initial = tf.truncated_normal(shape, stddev=0.1)
W_conv1a = tf.Variable(initial)
initial = tf.constant(0.1, shape=shape)
b_conv1a = tf.Variable(initial)
h_conv1a = tf.nn.relu(conv3d(x_image, W_conv1a) + b_conv1a)
tf.nn.conv3d(x, W, strides=[1, 1, 1, 1, 1], padding='SAME')