Skip to content

Instantly share code, notes, and snippets.

@sadimanna
Last active October 8, 2020 09:53
Show Gist options
  • Save sadimanna/c754e879da05e20e85cadb099e3f724f to your computer and use it in GitHub Desktop.
Save sadimanna/c754e879da05e20e85cadb099e3f724f to your computer and use it in GitHub Desktop.
img_input = Input(shape=(256,256,3))
x = conv2d(img_input,32,3,2,'valid',True,name='conv1')
x = conv2d(x,32,3,1,'valid',True,name='conv2')
x = conv2d(x,64,3,1,'valid',True,name='conv3')
x_11 = MaxPooling2D(3,strides=1,padding='valid',name='stem_br_11'+'_maxpool_1')(x)
x_12 = conv2d(x,64,3,1,'valid',True,name='stem_br_12')
x = Concatenate(axis=3, name = 'stem_concat_1')([x_11,x_12])
x_21 = conv2d(x,64,1,1,'same',True,name='stem_br_211')
x_21 = conv2d(x_21,64,[1,7],1,'same',True,name='stem_br_212')
x_21 = conv2d(x_21,64,[7,1],1,'same',True,name='stem_br_213')
x_21 = conv2d(x_21,96,3,1,'valid',True,name='stem_br_214')
x_22 = conv2d(x,64,1,1,'same',True,name='stem_br_221')
x_22 = conv2d(x_22,96,3,1,'valid',True,name='stem_br_222')
x = Concatenate(axis=3, name = 'stem_concat_2')([x_21,x_22])
x_31 = conv2d(x,192,3,1,'valid',True,name='stem_br_31')
x_32 = MaxPooling2D(3,strides=1,padding='valid',name='stem_br_32'+'_maxpool_2')(x)
x = Concatenate(axis=3, name = 'stem_concat_3')([x_31,x_32])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment