wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-08-17 11:20:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improve auxilary head

Browse Source
This commit is contained in:
Somshubra Majumdar
2017-11-18 13:38:12 -06:00
parent 4a99de40a2
commit 09ec44fa80
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/applications/nasnet.py
+5 -3
View File
@@ -195,7 +195,8 @@ def NASNet(input_shape=None,
auxilary_x = None
if use_auxilary_branch:
img_dim = 2 if K.image_data_format() == 'channels_first' else -2
img_height = 1 if K.image_data_format() == 'channels_first' else 2
img_width = 2 if K.image_data_format() == 'channels_first' else 3
with K.name_scope('auxilary_branch'):
auxilary_x = Activation('relu')(x)
@@ -206,8 +207,9 @@ def NASNet(input_shape=None,
name='aux_bn_projection')(auxilary_x)
auxilary_x = Activation('relu')(auxilary_x)
auxilary_x = Conv2D(768, auxilary_x._keras_shape[img_dim], padding='valid', use_bias=False,
kernel_initializer='he_normal', name='aux_conv_reduction')(auxilary_x)
auxilary_x = Conv2D(768, (auxilary_x._keras_shape[img_height], auxilary_x._keras_shape[img_width]),
padding='valid', use_bias=False, kernel_initializer='he_normal',
name='aux_conv_reduction')(auxilary_x)
auxilary_x = BatchNormalization(axis=channel_dim, momentum=_BN_DECAY, epsilon=_BN_EPSILON,
name='aux_bn_reduction')(auxilary_x)
auxilary_x = Activation('relu')(auxilary_x)