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

ai: how did I miss that

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-01 18:07:53 +01:00
parent b2f1ba29bb
commit 8a86728b54
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -16,19 +16,20 @@ def model_rainfallwater_contrastive(shape_rainfall, shape_water, batch_size=64,
water_channels = 1 # added in dataset → make_dataset → parse_item water_channels = 1 # added in dataset → make_dataset → parse_item
input_rainfall = tf.keras.layers.Input( input_rainfall = tf.keras.layers.Input(
shape=shape_rainfall shape=(rainfall_width, rainfall_height, rainfall_channels)
) )
input_water = tf.keras.layers.Input( input_water = tf.keras.layers.Input(
shape=(water_width, water_height, water_channels) shape=(water_width, water_height, water_channels)
) )
print("MAKE ENCODER rainfall")
rainfall = LayerContrastiveEncoder( rainfall = LayerContrastiveEncoder(
input_width=rainfall_width, input_width=rainfall_width,
input_height=rainfall_height, input_height=rainfall_height,
channels=rainfall_channels, channels=rainfall_channels,
feature_dim=feature_dim feature_dim=feature_dim
)(input_rainfall) )(input_rainfall)
print("MAKE ENCODER water")
water = LayerContrastiveEncoder( water = LayerContrastiveEncoder(
input_width=water_width, input_width=water_width,
input_height=water_height, input_height=water_height,