Advertisement
Guest User

Untitled

a guest
Feb 22nd, 2018
72
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 2.02 KB | None | 0 0
  1.     # Odessa Version
  2.     # Modified version of HTW approach
  3.     def _steering_onpu(self):
  4.         model = Sequential()
  5.         model.add(BatchNormalization(input_shape=(96,128,3)))
  6.  
  7.         # Convolution no.1
  8.         model.add(
  9.             Conv2D(
  10.                 filters=16,
  11.                 kernel_size=(5, 5),
  12.                 strides=(2, 2),
  13.                 bias_initializer='he_normal',
  14.                 padding="valid"))
  15.         model.add(ELU())
  16.  
  17.         model.add(MaxPooling2D(pool_size=(2, 2)))
  18.         model.add(BatchNormalization())
  19.  
  20.         # Convolution no.2
  21.         model.add(
  22.             Conv2D(
  23.                 filters=24,
  24.                 kernel_size=(5, 5),
  25.                 strides=(2, 2),
  26.                 bias_initializer='he_normal',
  27.                 padding="valid"))
  28.         model.add(ELU())
  29.         model.add(BatchNormalization())
  30.  
  31.         # Convolution no.3
  32.         model.add(
  33.             Conv2D(
  34.                 filters=32,
  35.                 kernel_size=(3, 3),
  36.                 bias_initializer='he_normal',
  37.                 padding="valid"))
  38.         model.add(ELU())
  39.         model.add(BatchNormalization())
  40.  
  41.         model.add(Flatten())
  42.  
  43.         # Fully connected no.1
  44.         model.add(
  45.             Dense(
  46.                 units=300,
  47.                 bias_initializer='he_normal'))
  48.         model.add(ELU())
  49.         model.add(BatchNormalization())
  50.         model.add(Dropout(0.25))
  51.  
  52.         # Fully connected no.2
  53.         model.add(
  54.             Dense(
  55.                 units=100,
  56.                 bias_initializer='he_normal'))
  57.         model.add(ELU())
  58.         model.add(BatchNormalization())
  59.         model.add(Dropout(0.25))
  60.  
  61.         # Fully connected no.3
  62.         model.add(
  63.             Dense(
  64.                 units=1,
  65.                 bias_initializer='he_normal',
  66.                 activation='tanh',
  67.                 name='prediction'))
  68.  
  69.         opt = optimizers.Nadam()
  70.         model.compile(loss='mse', optimizer=opt, metrics=['mse'])
  71.         model.summary()
  72.         return model
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement