mirror of
https://github.com/prise6/smart-iss-posts
synced 2024-04-27 11:31:51 +02:00
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from iss.models.AbstractModel import AbstractModel
|
|
from keras.layers import Input, Dense, Conv2D, MaxPooling2D, UpSampling2D, Reshape, Flatten
|
|
from keras.optimizers import Adadelta
|
|
from keras.models import Model
|
|
import numpy as np
|
|
|
|
class SimpleAutoEncoder(AbstractModel):
|
|
|
|
def __init__(self, config):
|
|
|
|
save_directory = config['save_directory']
|
|
model_name = config['model_name']
|
|
|
|
super().__init__(save_directory, model_name)
|
|
|
|
self.activation = config['activation']
|
|
self.input_shape = (config['input_height'], config['input_width'], config['input_channel'])
|
|
self.lr = config['learning_rate']
|
|
self.build_model()
|
|
|
|
def build_model(self):
|
|
input_shape = self.input_shape
|
|
|
|
picture = Input(shape = input_shape)
|
|
|
|
x = Flatten()(picture)
|
|
layer_1 = Dense(2000, activation = 'relu', name = 'enc_1')(x)
|
|
layer_2 = Dense(100, activation = 'relu', name = 'enc_2')(layer_1)
|
|
layer_3 = Dense(30, activation = 'relu', name = 'enc_3')(layer_2)
|
|
layer_4 = Dense(100, activation = 'relu', name = 'dec_1')(layer_3)
|
|
layer_5 = Dense(2000, activation = 'relu', name = 'dec_2')(layer_4)
|
|
|
|
# encoded network
|
|
# x = Conv2D(1, (3, 3), activation = 'relu', padding = 'same', name = 'enc_conv_1')(picture)
|
|
# encoded = MaxPooling2D((2, 2))(x)
|
|
|
|
# decoded network
|
|
# x = Conv2D(1, (3, 3), activation = 'relu', padding = 'same', name = 'dec_conv_1')(encoded)
|
|
# x = UpSampling2D((2, 2))(x)
|
|
# x = Flatten()(x)
|
|
x = Dense(np.prod(input_shape), activation = self.activation)(layer_5)
|
|
decoded = Reshape((input_shape))(x)
|
|
|
|
self.model = Model(picture, decoded)
|
|
|
|
optimizer = Adadelta(lr = self.lr, rho = 0.95, epsilon = None, decay = 0.0)
|
|
|
|
self.model.compile(optimizer = optimizer, loss = 'binary_crossentropy')
|