From 9384b89165f1177d2c470a009aac6c43e35e5d53 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 10 Nov 2022 20:53:37 +0000 Subject: [PATCH] =?UTF-8?q?model=5Fsegmentation:=20spare=20=E2=86=92=20nor?= =?UTF-8?q?mal=20crossentropy,=20activation=20functions=20at=20end?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- aimodel/src/lib/ai/model_rainfallwater_segmentation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py index ce8977f..becd9fd 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py +++ b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py @@ -44,8 +44,8 @@ def model_rainfallwater_segmentation(metadata, shape_water_out, model_arch="conv # TODO: An attention layer here instead of a dense layer, with a skip connection perhaps? logger.warning("Warning: TODO implement attention from https://ieeexplore.ieee.org/document/9076883") - layer_next = tf.keras.layers.Dense(32)(layer_next) - layer_next = tf.keras.layers.Conv2D(water_bins, kernel_size=1, padding="same")(layer_next) + layer_next = tf.keras.layers.Dense(32, activation="relu")(layer_next) + layer_next = tf.keras.layers.Conv2D(water_bins, activation="relu", kernel_size=1, padding="same")(layer_next) layer_next = tf.keras.layers.Softmax(axis=-1)(layer_next) model = tf.keras.Model(