From b5e08f92fe4cad1db5c260b93a84eb77e4efb1d8 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 28 Sep 2022 18:14:09 +0100 Subject: [PATCH] the long night continues --- aimodel/src/lib/ai/model_rainfallwater_segmentation.py | 2 +- aimodel/src/lib/dataset/dataset_segmenter.py | 2 +- 2 files 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 87b4bab..f2b23da 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_segmentation.py +++ b/aimodel/src/lib/ai/model_rainfallwater_segmentation.py @@ -6,7 +6,7 @@ import tensorflow as tf from .components.convnext_inverse import do_convnext_inverse def model_rainfallwater_segmentation(metadata, feature_dim_in, shape_water_out, batch_size=64, summary_file=None): - out_water_width, out_water_height, out_water_channels = shape_water_out + out_water_width, out_water_height = shape_water_out layer_input = tf.keras.layers.Input( diff --git a/aimodel/src/lib/dataset/dataset_segmenter.py b/aimodel/src/lib/dataset/dataset_segmenter.py index d40e84a..a864a4b 100644 --- a/aimodel/src/lib/dataset/dataset_segmenter.py +++ b/aimodel/src/lib/dataset/dataset_segmenter.py @@ -15,7 +15,7 @@ from .shuffle import shuffle # TO PARSE: def parse_item(metadata, shape_water_desired, water_threshold=0.1): - water_width_source, water_height_source = metadata["waterdepth"] + water_width_source, water_height_source, _water_channels_source = metadata["waterdepth"] water_width_target, water_height_target = shape_water_desired water_offset_x = math.ceil((water_width_source - water_width_target) / 2) water_offset_y = math.ceil((water_height_source - water_height_target) / 2)