Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- input_layer = Input(shape=(258, 540, 1))
- # encoder
- encoder = Conv2D(64, (3, 3), activation='relu', padding='same')(input_layer)
- encoder = MaxPooling2D((2, 2), padding='same')(encoder)
- # decoder
- decoder = Conv2D(64, (3, 3), activation='relu', padding='same')(encoder)
- decoder = UpSampling2D((2, 2))(decoder)
- output_layer = Conv2D(1, (3, 3), activation='sigmoid', padding='same')(decoder)
- ae = Model(input_layer, output_layer)
- ae.compile(loss='mse', optimizer=Adam(lr=0.001))
- batch_size = 16
- epochs = 200
- early_stopping = EarlyStopping(monitor='val_loss',min_delta=0,patience=5,verbose=1, mode='auto')
- history = ae.fit(x_train, y_train, batch_size=batch_size, epochs=epochs, validation_data=(x_val, y_val), callbacks=[early_stopping])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement