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

dir: also plot inputs

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-03-09 19:13:25 +00:00
parent 436ab78438
commit ad52ae9241
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/deeplabv3_plus_test_rainfall.py
View file

@ -313,6 +313,8 @@ def plot_predictions(filepath, input_items, colormap, model):
filepath.replace("$$", str(i)),
[
# input_tensor,
tf.math.reduce_max(input_pair[0][:,:,:-1], axis=-1), # rainfall only
input_pair[0][:,:,-1], # heightmap
input_pair[1], #label_colourmap,
prediction_mask[:,:,1],
prediction_colormap