sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-24 18:23:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

ds_eo: rogue variables

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-01-17 19:03:14 +00:00
parent 64c57bbc21
commit b9bea26d26
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 0 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -16,9 +16,6 @@ from .parse_heightmap import parse_heightmap
# TO PARSE: # TO PARSE:
def parse_item(metadata, water_threshold=0.1, water_bins=2, heightmap=None, rainfall_scale_up=1, windowsize=33): def parse_item(metadata, water_threshold=0.1, water_bins=2, heightmap=None, rainfall_scale_up=1, windowsize=33):
if input_size == "same":
input_size = output_size # This is almost always the case with e.g. the DeepLabV3+ model
water_height_source, water_width_source = metadata["waterdepth"] water_height_source, water_width_source = metadata["waterdepth"]
rainfall_channels, rainfall_height_source, rainfall_width_source = metadata["rainfallradar"] rainfall_channels, rainfall_height_source, rainfall_width_source = metadata["rainfallradar"]
@ -29,8 +26,6 @@ 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 shape", metadata["waterdepth"])
print("DEBUG DATASET:water_threshold", water_threshold) print("DEBUG DATASET:water_threshold", water_threshold)
print("DEBUG DATASET:water_bins", water_bins) print("DEBUG DATASET:water_bins", water_bins)
print("DEBUG DATASET:output_size", output_size)
print("DEBUG DATASET:input_size", input_size)
if heightmap is not None: if heightmap is not None:
heightmap = tf.expand_dims(heightmap, axis=-1) heightmap = tf.expand_dims(heightmap, axis=-1)