This commit is contained in:
Starbeamrainbowlabs 2022-09-02 18:45:23 +01:00
parent 6135bcd0cd
commit 1c5defdcd6
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -1,3 +1,4 @@
import math
from curses import meta from curses import meta
from loguru import logger from loguru import logger
@ -13,7 +14,7 @@ def model_rainfallwater_contrastive(metadata, shape_water, batch_size=64, featur
water_width, water_height = shape_water # shape = [width, height] water_width, water_height = shape_water # shape = [width, height]
water_channels = 1 # added in dataset → make_dataset → parse_item water_channels = 1 # added in dataset → make_dataset → parse_item
rainfall_width, rainfall_height = rainfall_width / 2, rainfall_height / 2 rainfall_width, rainfall_height = math.floor(rainfall_width / 2), math.floor(rainfall_height / 2)
logger.info("SOURCE shape_rainfall " + str(metadata["rainfallradar"])) logger.info("SOURCE shape_rainfall " + str(metadata["rainfallradar"]))
logger.info("SOURCE shape_water " + str(metadata["waterdepth"])) logger.info("SOURCE shape_water " + str(metadata["waterdepth"]))