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

dlr: make size fo prediction graph actually sensible

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-03-09 19:54:27 +00:00
parent 8d932757b5
commit d24381babb
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
aimodel/src/deeplabv3_plus_test_rainfall.py
View file

@ -4,10 +4,12 @@
from datetime import datetime
from loguru import logger
from lib.ai.helpers.summarywriter import summarywriter
from lib.ai.components.CallbackCustomModelCheckpoint import CallbackCustomModelCheckpoint
import os
import math
import cv2
import numpy as np
from glob import glob
@ -293,8 +295,9 @@ def get_overlay(image, coloured_mask):
def plot_samples_matplotlib(filepath, display_list):
plt.figure(figsize=(16, 8))
for i in range(len(display_list)):
plt.subplot(1, len(display_list), i+1)
plt.subplot(2, math.ceil(len(display_list) / 2), i+1)
if display_list[i].shape[-1] == 3:
plt.imshow(tf.keras.preprocessing.image.array_to_img(display_list[i]))
else: