From e51087d0a99625defd107cc0629b10c32368d9ee Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 28 Sep 2022 18:22:48 +0100 Subject: [PATCH] add reshape layer --- aimodel/src/lib/ai/model_rainfallwater_segmentation.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py index 92ae012..6e76164 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py +++ b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py @@ -15,8 +15,10 @@ 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.LayerNormalization(name="stage_begin.norm", epsilon=1e-6)(layer_next) - layer_next = tf.keras.layers.ReLU(name="stage_begin.relu")(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 = do_convnext_inverse(layer_next, arch_name="convnext_i_tiny")