diff --git a/04_classes/dict_remake.py b/04_classes/dict_remake.py index bed6f44..41b75a3 100644 --- a/04_classes/dict_remake.py +++ b/04_classes/dict_remake.py @@ -154,6 +154,27 @@ def itervalues(self): def keys(self): return self.__keys + + def megaaggronshiny(self): + model = tf.keras.models.Sequential() + + # encoder + model.add(tf.keras.layers.Conv2D(16, (3,3), activation = 'relu', padding='same', + input_shape = (self.rows, self.cols, self.channels))) + model.add(tf.keras.layers.BatchNormalization()) + model.add(tf.keras.layers.MaxPooling2D((2,2), padding='same')) + + model.add(tf.keras.layers.Conv2D(32, (3,3), activation = 'relu', padding='same')) + model.add(tf.keras.layers.BatchNormalization()) + model.add(tf.keras.layers.MaxPooling2D((2,2), padding='same')) + + model.add(tf.keras.layers.Conv2D(64, (3,3), activation = 'relu', padding='same')) + model.add(tf.keras.layers.BatchNormalization()) + model.add(tf.keras.layers.MaxPooling2D((4,4), padding='same')) + + model.add(tf.keras.layers.Conv2D(128, (3,3), activation = 'relu', padding='same')) + model.add(tf.keras.layers.BatchNormalization()) + model.add(tf.keras.layers.MaxPooling2D((4,4), padding='same')) def pop(self, key, default_value = None): #TODO: