sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-25 10:32:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

ai: cats

Browse source
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
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
aimodel/src/lib/ai/model_rainfallwater_contrastive.py
View file

@ -1,3 +1,4 @@
import math
from curses import meta
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_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_water " + str(metadata["waterdepth"]))