From a3787f06479c24d68809b214dadb5dd4244ac7bb Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 20 Jan 2023 18:34:56 +0000 Subject: [PATCH] debug --- aimodel/src/lib/dataset/dataset_encoderonly.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/aimodel/src/lib/dataset/dataset_encoderonly.py b/aimodel/src/lib/dataset/dataset_encoderonly.py index 32a42df..de5b16d 100644 --- a/aimodel/src/lib/dataset/dataset_encoderonly.py +++ b/aimodel/src/lib/dataset/dataset_encoderonly.py @@ -81,7 +81,9 @@ def parse_item(metadata, water_threshold=0.1, water_bins=2, heightmap=None, rain 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), + rainfall = tf.expand_dims(rainfall, axis=0) + print("DEBUG:dataset BEFORE_PATCHES rainfall", rainfall.shape) + rainfall = tf.image.extract_patches(rainfall, sizes=[1,windowsize,windowsize,1], strides=[1,1,1,1], rates=[1,1,1,1],