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

dlr predict: output with higher quality

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-01-12 18:43:48 +00:00
parent 1b5bb14d8f
commit 0d41bbba94
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
aimodel/src/deeplabv3_plus_test_rainfall.py
View file

@ -245,14 +245,14 @@ def get_overlay(image, coloured_mask):
return overlay return overlay
def plot_samples_matplotlib(filepath, display_list, figsize=(5, 3)): def plot_samples_matplotlib(filepath, display_list):
for i in range(len(display_list)): for i in range(len(display_list)):
plt.subplot(1, len(display_list), i+1) plt.subplot(1, len(display_list), i+1)
if display_list[i].shape[-1] == 3: if display_list[i].shape[-1] == 3:
plt.imshow(tf.keras.preprocessing.image.array_to_img(display_list[i])) plt.imshow(tf.keras.preprocessing.image.array_to_img(display_list[i]))
else: else:
plt.imshow(display_list[i]) plt.imshow(display_list[i])
plt.savefig(filepath) plt.savefig(filepath, dpi=200)
def plot_predictions(filepath, input_items, colormap, model): def plot_predictions(filepath, input_items, colormap, model):
@ -270,8 +270,7 @@ def plot_predictions(filepath, input_items, colormap, model):
# input_tensor, # input_tensor,
input_pair[1], #label_colourmap input_pair[1], #label_colourmap
prediction_colormap prediction_colormap
], ]
figsize=(18, 14)
) )
i += 1 i += 1