Advertisement
Guest User

Untitled

a guest
Jun 20th, 2018
68
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.92 KB | None | 0 0
  1. def model_pass(input):
  2.     input_layer = tf.to_float(input)
  3.     input_layer_norm = (2 / 255) * input_layer - 1
  4.  
  5.       # Convolutional Layer #1
  6.     conv1 = tf.layers.conv2d(
  7.       inputs=input_layer_norm,
  8.       filters=32,
  9.       kernel_size=[3, 3],
  10.       strides = [2,2],
  11.       padding="same",
  12.       activation=tf.nn.relu,
  13.       kernel_initializer=tf.contrib.layers.xavier_initializer())
  14.  
  15.     conv2 = tf.layers.conv2d(
  16.       inputs=conv1,
  17.       filters=64,
  18.       kernel_size=[3, 3],
  19.       strides = [2,2],
  20.       padding="same",
  21.       activation=tf.nn.relu,
  22.       kernel_initializer=tf.contrib.layers.xavier_initializer())
  23.  
  24.     conv3 = tf.layers.conv2d(
  25.         inputs=conv2,
  26.         filters=128,
  27.         kernel_size=[3, 3],
  28.         strides = [2,2],
  29.         padding="same",
  30.         activation=tf.nn.relu,
  31.         kernel_initializer=tf.contrib.layers.xavier_initializer())
  32.  
  33.     conv4 = tf.layers.conv2d(
  34.         inputs=conv3,
  35.         filters=256,
  36.         kernel_size=[3, 3],
  37.         strides = [2,2],
  38.         padding="same",
  39.         activation=tf.nn.relu,
  40.         kernel_initializer=tf.contrib.layers.xavier_initializer())
  41.  
  42.     conv5 = tf.layers.conv2d(
  43.         inputs=conv4,
  44.         filters=512,
  45.         kernel_size=[3, 3],
  46.         strides = [2,2],
  47.         padding="same",
  48.         activation=tf.nn.relu,
  49.         kernel_initializer=tf.contrib.layers.xavier_initializer())
  50.  
  51.  
  52.     conv6 = tf.layers.conv2d(
  53.         inputs=conv5,
  54.         filters=1024,
  55.         kernel_size=[3, 3],
  56.         strides = [2,2],
  57.         padding="same",
  58.         activation=tf.nn.relu,
  59.         kernel_initializer=tf.contrib.layers.xavier_initializer())
  60.  
  61.     pool = tf.layers.max_pooling2d(inputs=conv6, pool_size=[5,5], strides=[1,1])
  62.     pool_flat = tf.reshape(pool, [-1, 1024])
  63.     dense = tf.layers.dense(inputs=pool_flat, units=1024, activation=tf.nn.relu)
  64.  
  65.     logits = tf.layers.dense(inputs=dense, units=2)
  66.    
  67.     return logits
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement