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

dataset_mono: fix sizing

Browse source 
it didn't account for rainfall_scale_up
This commit is contained in:
Starbeamrainbowlabs 2023-01-05 18:38:47 +00:00
parent 19bb2fcac0
commit aca7b83a78
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -24,10 +24,12 @@ def parse_item(metadata, output_size=100, input_size="same", water_threshold=0.1
water_offset_y = math.ceil((water_height_source - output_size) / 2) water_offset_y = math.ceil((water_height_source - output_size) / 2)
_, rainfall_height_source, rainfall_width_source = metadata["rainfallradar"] _, rainfall_height_source, rainfall_width_source = metadata["rainfallradar"]
rainfall_height_source *= rainfall_scale_up
rainfall_width_source *= rainfall_scale_up
rainfall_offset_x = math.ceil((rainfall_width_source - input_size) / 2) rainfall_offset_x = math.ceil((rainfall_width_source - input_size) / 2)
rainfall_offset_y = math.ceil((rainfall_height_source - input_size) / 2) rainfall_offset_y = math.ceil((rainfall_height_source - input_size) / 2)
print("DEBUG DATASET:rainfall shape", metadata["rainfallradar"]) print("DEBUG DATASET:rainfall shape", metadata["rainfallradar"], "/", f"{rainfall_width_source} h{rainfall_height_source}")
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)