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

dlr: fix crash

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

3
aimodel/src/deeplabv3_plus_test_rainfall.py
View file

@ -314,7 +314,8 @@ def save_samples(filepath, save_list):
def plot_predictions(filepath, input_items, colormap, model):
filepath_jsonl = filepath.replace("_$$", "").replace(".png", ".jsonl")
os.truncate(filepath_jsonl, 0)
if os.path.exists(filepath_jsonl):
os.truncate(filepath_jsonl, 0)
i = 0
for input_pair in input_items: