sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-21 17:03:00 +00:00
Code Issues Projects Releases Packages Wiki Activity

ds eo: debug

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-01-20 18:36:14 +00:00
parent a3787f0647
commit 5b54ceec48
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -26,6 +26,7 @@ def parse_item(metadata, water_threshold=0.1, water_bins=2, heightmap=None, rain
print("DEBUG DATASET:water shape", metadata["waterdepth"])
print("DEBUG DATASET:water_threshold", water_threshold)
print("DEBUG DATASET:water_bins", water_bins)
print("DEBUG DATASET:rainfall_scale_up", rainfall_scale_up)
if heightmap is not None:
heightmap = tf.expand_dims(heightmap, axis=-1)