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

autoset the value of factor

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-05-04 19:57:02 +01:00
parent 214afcc914
commit ac040717e6
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/deeplabv3_plus_test_rainfall.py
View file

@ -160,8 +160,9 @@ if PATH_CHECKPOINT is None:
x = backbone.get_layer("conv4_block6_2_relu").output
x = DilatedSpatialPyramidPooling(x)
factor = 4 if upsample == 2 else 8 # else: upsample == 1. other values are not supported yet because maths
input_a = tf.keras.layers.UpSampling2D(
size=(image_size // 4 // x.shape[1] * 2, image_size // 4 // x.shape[2] * 2), # <--- UPSAMPLE after pyramid
size=(image_size // factor // x.shape[1] * 2, image_size // factor // x.shape[2] * 2), # <--- UPSAMPLE after pyramid
interpolation="bilinear",
)(x)
input_b = backbone.get_layer("conv2_block3_2_relu").output