Guest User

tensorflow_code

a guest
Apr 2nd, 2018
48
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 2.28 KB | None | 0 0
  1. images_placeholder = tf.placeholder(tf.float32, [None, training_images.shape[1], training_images.shape[2], 1], name = "input_node")
  2. one_hot_placeholder = tf.placeholder(tf.float32, [None, 3], name = "one_hot_node")
  3. features_placeholder = tf.placeholder(tf.float32,[None, 4], name = "features_node")
  4. total_placeholder = tf.placeholder(tf.float32,[None,7], name = "total_node")
  5.  
  6. def convolution_layer(input, filter_size, stride, filters_number, reg_value, layer_number):
  7.  
  8.     layer_name = "convolution" + str(layer_number)
  9.  
  10.     conv_output = tf.layers.conv2d(
  11.     input,
  12.     filters_number,
  13.     filter_size,
  14.     strides=stride,
  15.     padding='valid',
  16.     data_format='channels_last',
  17.     dilation_rate=(1, 1),
  18.     activation=tf.nn.relu,
  19.     use_bias=True,
  20.     kernel_initializer=tf.contrib.layers.xavier_initializer(),
  21.     bias_initializer=tf.zeros_initializer(),
  22.     kernel_regularizer=tf.contrib.layers.l2_regularizer(reg_value),
  23.     bias_regularizer=None,
  24.     activity_regularizer=None,
  25.     kernel_constraint=None,
  26.     bias_constraint=None,
  27.     trainable=True,
  28.     name=layer_name,
  29.     reuse=None)
  30.  
  31.     return conv_output
  32.  
  33. def fully_connected_layers(input, neurons_number, layer_number):
  34.     layer_name = "fully_connected" + str(layer_number)
  35.     full_con_output = tf.layers.dense(
  36.         inputs=input,
  37.         units = neurons_number,
  38.         activation = tf.nn.relu,
  39.         kernel_initializer = tf.contrib.layers.xavier_initializer(),
  40.         kernel_regularizer = tf.contrib.layers.l2_regularizer(0.001),
  41.         name=layer_name)
  42.  
  43.     return full_con_output
  44.  
  45. #architecture here
  46. with tf.name_scope('Conv1'):
  47.     conv1 = convolution_layer( images_placeholder, 5, 2, 6, 0.001, 1 )
  48.     pool1 = tf.nn.max_pool(conv1, [1,2,2,1], [1,2,2,1], 'SAME', data_format='NHWC', name = "pooling_1")
  49. with tf.name_scope('Conv2'):
  50.     conv2 = convolution_layer( pool1, 11, 2, 6, 0.001, 2 )
  51.     pool2 = tf.nn.max_pool(conv2, [1,2,2,1], [1,2,2,1], 'SAME', data_format='NHWC', name = "pooling_2")
  52.     shape = pool2.get_shape()
  53.     flatten = tf.reshape( pool2, [ -1, 216 ] )
  54. with tf.name_scope('Dense2'):
  55.     dense2 = fully_connected_layers( flatten, 40, 4 )
  56. with tf.name_scope('Dense4'):
  57.     dense4 = fully_connected_layers( dense2, 4, 6 )
  58.     features_output = tf.nn.relu( dense4, "output_node" )
Add Comment
Please, Sign In to add comment