aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lenet.py16
1 files changed, 15 insertions, 1 deletions
diff --git a/lenet.py b/lenet.py
index 97479ed..71b1c9e 100644
--- a/lenet.py
+++ b/lenet.py
@@ -57,7 +57,7 @@ def get_lenet(shape):
model = keras.Sequential()
model.add(Conv2D(filters=6, kernel_size=(3, 3), activation='relu', input_shape=shape))
model.add(AveragePooling2D())
-
+
model.add(Conv2D(filters=16, kernel_size=(3, 3), activation='relu'))
model.add(AveragePooling2D())
model.add(Flatten())
@@ -68,6 +68,20 @@ def get_lenet(shape):
model.add(Dense(units=10, activation = 'relu'))
return model
+def get_lenet_icp(shape):
+ model = keras.Sequential()
+ model.add(Conv2D(filters=6, kernel_size=(3, 3), activation='relu', input_shape=(32,32,1)))
+ model.add(AveragePooling2D())
+
+ model.add(Conv2D(filters=16, kernel_size=(3, 3), activation='relu'))
+ model.add(AveragePooling2D())
+ model.add(Flatten())
+
+ model.add(Dense(units=120, activation='relu'))
+ model.add(Dense(units=84, activation='relu'))
+ model.add(Dense(units=10, activation = 'relu'))
+ return model
+
def plot_history(history, metric = None):
# Plots the loss history of training and validation (if existing)
# and a given metric