From 058e3b62488c9546ba98add200bb844cc16e0fb1 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Mon, 3 Oct 2022 17:31:36 +0100 Subject: [PATCH] =?UTF-8?q?model=5Fsegmentation:=20cast=20float=20?= =?UTF-8?q?=E2=86=92=20int?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- aimodel/src/lib/ai/model_rainfallwater_segmentation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py index 95fb646..f46897c 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py +++ b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py @@ -18,7 +18,7 @@ def model_rainfallwater_segmentation(metadata, feature_dim_in, shape_water_out, 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.Reshape((4, 4, feature_dim_in/(4*4)), name="cns.stable_begin.reshape")(layer_next) + layer_next = tf.keras.layers.Reshape((4, 4, math.floor(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)