From 9bd5e0f7a3c314b9f4a3ffd8f353d2089310924c Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 20 Jan 2023 17:25:22 +0000 Subject: [PATCH] ds_eo: debug there's something fishy going on here. --- aimodel/src/lib/dataset/dataset_encoderonly.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/aimodel/src/lib/dataset/dataset_encoderonly.py b/aimodel/src/lib/dataset/dataset_encoderonly.py index e50aea9..32a42df 100644 --- a/aimodel/src/lib/dataset/dataset_encoderonly.py +++ b/aimodel/src/lib/dataset/dataset_encoderonly.py @@ -79,6 +79,7 @@ def parse_item(metadata, water_threshold=0.1, water_bins=2, heightmap=None, rain # water = tf.one_hot(water, water_bins, axis=-1, dtype=tf.int32) # LOSS dice water = tf.cast(tf.math.greater_equal(water, water_threshold), dtype=tf.int32) + print("DEBUG:dataset AFTER_BINARISE water", water.shape) rainfall = tf.image.extract_patches(tf.expand_dims(rainfall, axis=0), sizes=[1,windowsize,windowsize,1], @@ -86,7 +87,10 @@ def parse_item(metadata, water_threshold=0.1, water_bins=2, heightmap=None, rain rates=[1,1,1,1], padding="VALID" ) + print("DEBUG:dataset AFTER_PATCHES rainfall", rainfall.shape, "windowsize", windowsize) rainfall = tf.reshape(rainfall, [-1, windowsize, windowsize, rainfall_channels]) + print("DEBUG:dataset AFTER_RESHAPE rainfall", + rainfall, "rainfall_channels", rainfall_channels) water = tf.reshape(water, [-1]) # we flatten because we cropped to the right shape above