Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #Encoder
- encoder_input = Input(shape=(256, 256, 1,))
- encoder_output = Conv2D(64, (3,3), activation='relu', padding='same', strides=2)(encoder_input)
- encoder_output = Conv2D(128, (3,3), activation='relu', padding='same')(encoder_output)
- encoder_output = Conv2D(128, (3,3), activation='relu', padding='same', strides=2)(encoder_output)
- encoder_output = Conv2D(256, (3,3), activation='relu', padding='same')(encoder_output)
- encoder_output = Conv2D(256, (3,3), activation='relu', padding='same', strides=2)(encoder_output)
- encoder_output = Conv2D(512, (3,3), activation='relu', padding='same')(encoder_output)
- encoder_output = Conv2D(512, (3,3), activation='relu', padding='same')(encoder_output)
- encoder_output = Conv2D(256, (3,3), activation='relu', padding='same')(encoder_output)
- #Decoder
- decoder_output = Conv2D(128, (3,3), activation='relu', padding='same')(encoder_output)
- decoder_output = UpSampling2D((2, 2))(decoder_output)
- decoder_output = Conv2D(64, (3,3), activation='relu', padding='same')(decoder_output)
- decoder_output = UpSampling2D((2, 2))(decoder_output)
- decoder_output = Conv2D(32, (3,3), activation='relu', padding='same')(decoder_output)
- decoder_output = Conv2D(16, (3,3), activation='relu', padding='same')(decoder_output)
- decoder_output = Conv2D(2, (3, 3), activation='tanh', padding='same')(decoder_output)
- decoder_output = UpSampling2D((2, 2))(decoder_output)
- model = Model(inputs=encoder_input, outputs=decoder_output)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement