From 7e529068a8f5c7c0010a2410e9e8389d667e4cd9 Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Fri, 12 Mar 2021 12:43:19 +0100
Subject: [PATCH] bigger kernel
---
mlair/model_modules/convolutional_networks.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/mlair/model_modules/convolutional_networks.py b/mlair/model_modules/convolutional_networks.py
index e7d1da23..0a16be7c 100644
--- a/mlair/model_modules/convolutional_networks.py
+++ b/mlair/model_modules/convolutional_networks.py
@@ -81,7 +81,7 @@ class CNN(AbstractModelClass):
Build the model.
"""
x_input = keras.layers.Input(shape=self._input_shape)
- kernel = (1, 1)
+ kernel = (5, 1)
pad_size = PadUtils.get_padding_for_same(kernel)
x_in = Padding2D("SymPad2D")(padding=pad_size, name="SymPad1")(x_input)
x_in = keras.layers.Conv2D(filters=16, kernel_size=kernel,
@@ -91,8 +91,8 @@ class CNN(AbstractModelClass):
x_in = keras.layers.Conv2D(filters=32, kernel_size=kernel,
kernel_initializer=self.kernel_initializer,
kernel_regularizer=self.kernel_regularizer)(x_in)
+ x_in = keras.layers.MaxPooling2D(kernel, strides=(1, 1), padding='valid')(x_in)
x_in = self.activation()(x_in)
- x_in = Padding2D("SymPad2D")(padding=pad_size, name="SymPad2")(x_in)
x_in = keras.layers.Conv2D(filters=64, kernel_size=kernel,
kernel_initializer=self.kernel_initializer,
kernel_regularizer=self.kernel_regularizer)(x_in)
--
GitLab