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: truncate jsonl before we start to avoid mixing things up

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-03-10 17:11:10 +00:00
parent 5fdf229d06
commit cf37aeb11a
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
aimodel/src/deeplabv3_plus_test_rainfall.py
View file

@ -9,6 +9,7 @@ from lib.ai.helpers.summarywriter import summarywriter
from lib.ai.components.CallbackCustomModelCheckpoint import CallbackCustomModelCheckpoint from lib.ai.components.CallbackCustomModelCheckpoint import CallbackCustomModelCheckpoint
import os import os
import io
import math import math
import cv2 import cv2
import numpy as np import numpy as np
@ -312,6 +313,9 @@ def save_samples(filepath, save_list):
handle.close() handle.close()
def plot_predictions(filepath, input_items, colormap, model): def plot_predictions(filepath, input_items, colormap, model):
filepath_jsonl = filepath.replace("_$$", "").replace(".png", ".jsonl")
os.truncate(filepath_jsonl, 0)
i = 0 i = 0
for input_pair in input_items: for input_pair in input_items:
prediction_mask = infer(image_tensor=input_pair[0], model=model) prediction_mask = infer(image_tensor=input_pair[0], model=model)
@ -334,7 +338,7 @@ def plot_predictions(filepath, input_items, colormap, model):
) )
save_samples( save_samples(
filepath.replace("_$$", "").replace(".png", ".jsonl"), filepath_jsonl,
prediction_mask prediction_mask
) )
i += 1 i += 1