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

fix crash 'cause we're only plotting 1 thing

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-01-11 17:41:55 +00:00
parent 3787155665
commit be7dd91f88
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

@ -247,6 +247,8 @@ def get_overlay(image, coloured_mask):
def plot_samples_matplotlib(filepath, display_list, figsize=(5, 3)): def plot_samples_matplotlib(filepath, display_list, figsize=(5, 3)):
_, axes = plt.subplots(nrows=1, ncols=len(display_list), figsize=figsize) _, axes = plt.subplots(nrows=1, ncols=len(display_list), figsize=figsize)
if not isinstance(axes, list):
axes = [ axes ]
for i in range(len(display_list)): for i in range(len(display_list)):
if display_list[i].shape[-1] == 3: if display_list[i].shape[-1] == 3:
axes[i].imshow(tf.keras.preprocessing.image.array_to_img(display_list[i])) axes[i].imshow(tf.keras.preprocessing.image.array_to_img(display_list[i]))