From 04e5ae0c45a022e56139f95a597a68c7864b2e10 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Mon, 3 Oct 2022 17:27:37 +0100 Subject: [PATCH] model_segmentation: redo reshape much cheese was applied :P --- .../src/lib/ai/model_rainfallwater_segmentation.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py index 6e76164..95fb646 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py +++ b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py @@ -14,11 +14,17 @@ def model_rainfallwater_segmentation(metadata, feature_dim_in, shape_water_out, ) # BEGIN - layer_next = tf.keras.layers.Dense(name="cns.stage.begin.dense", units=feature_dim_in)(layer_input) + layer_next = tf.keras.layers.Dense(name="cns.stage.begin.dense1", units=feature_dim_in)(layer_input) + layer_next = tf.keras.layers.ReLU(name="cns.stage_begin.relu1")(layer_next) layer_next = tf.keras.layers.LayerNormalization(name="cns.stage_begin.norm", epsilon=1e-6)(layer_next) - layer_next = tf.keras.layers.ReLU(name="cns.stage_begin.relu")(layer_next) - layer_next = tf.keras.layers.Reshape((1, 1, feature_dim_in), name="cns.stable_begin.reshape")(layer_next) + layer_next = tf.keras.layers.Reshape((4, 4, feature_dim_in/(4*4)), name="cns.stable_begin.reshape")(layer_next) + layer_next = tf.keras.layers.Dense(name="cns.stage.begin.dense2", units=feature_dim_in)(layer_next) + layer_next = tf.keras.layers.ReLU(name="cns.stage_begin.relu2")(layer_next) + layer_next = tf.keras.layers.LayerNormalization(name="cns.stage_begin.norm", epsilon=1e-6)(layer_next) + + + # layer_next = tf.keras.layers.Reshape((1, 1, feature_dim_in), name="cns.stable_begin.reshape")(layer_next) layer_next = do_convnext_inverse(layer_next, arch_name="convnext_i_tiny")