TensorFlow实现去噪自编码器（Denoising Autoencoder）

kernel_size = 3

latent_dim = 16

layer_filters = [32,64]

“”"

模型

“”"

#编码器

inputs = keras.layers.Input(shape=input_shape,name=‘encoder_input’)

x = inputs

for filters in layer_filters:

x = keras.layers.Conv2D(filters=filters,

kernel_size=kernel_size,

strides=2,

activation=‘relu’,

padding=‘same’)(x)

shape = keras.backend.int_shape(x)

x = keras.layers.Flatten()(x)

latent = keras.layers.Dense(latent_dim,name=‘latent_vector’)(x)

encoder = keras.Model(inputs,latent,name=‘encoder’)

encoder.summary()

解码器

latent_inputs = keras.layers.Input(shape=(latent_dim,),name=‘decoder_input’)

x = keras.layers.Dense(shape[1]*shape[2]*shape[3])(latent_inputs)

x = keras.layers.Reshape((shape[1],shape[2],shape