Skip to content

Instantly share code, notes, and snippets.

@PranjalDureja0002
Created May 7, 2021 12:53
Show Gist options
  • Save PranjalDureja0002/fe478939ad51912dae7470c3cd4ac3d7 to your computer and use it in GitHub Desktop.
Save PranjalDureja0002/fe478939ad51912dae7470c3cd4ac3d7 to your computer and use it in GitHub Desktop.
model
def fun_column(x,output):
x = Conv2D(128, (1, 1), activation = 'relu', name='conv7_column')(x)
x = Dropout(0.8, name='block7_dropout_1')(x)
concatenated = Concatenate()([x,output])
concatenated = Concatenate()([concatenated,output])
x = UpSampling2D(size=(2, 2))(concatenated)
x = UpSampling2D(size=(2, 2))(x)
x = UpSampling2D(size=(2, 2))(x)
x = UpSampling2D(size=(2, 2))(x)
last = tf.keras.layers.Conv2DTranspose(3, 3, strides=2,padding='same', name='column_output')
x = last(x)
return x
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment