sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-12-22 22:25:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

ds_eo: debug

Browse source 
there's something fishy going on here.
This commit is contained in:
Starbeamrainbowlabs 2023-01-20 17:25:22 +00:00
parent aa3831bc4f
commit 9bd5e0f7a3
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
aimodel/src/lib/dataset/dataset_encoderonly.py
View file

@ -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