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: fix predictions

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-01-10 19:19:30 +00:00
parent 2591cbe6bc
commit 0e3de8f5fc
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
aimodel/src/deeplabv3_plus_test_rainfall.py
View file

@ -252,10 +252,10 @@ def plot_predictions(filepath, input_items, colormap, model):
for input_tensor in input_items: for input_tensor in input_items:
prediction_mask = infer(image_tensor=input_tensor, model=model) prediction_mask = infer(image_tensor=input_tensor, model=model)
prediction_colormap = decode_segmentation_masks(prediction_mask, colormap, 20) prediction_colormap = decode_segmentation_masks(prediction_mask, colormap, 20)
overlay = get_overlay(input_tensor, prediction_colormap)
plot_samples_matplotlib( plot_samples_matplotlib(
filepath, filepath,
[input_tensor, overlay, prediction_colormap], [input_tensor, prediction_colormap],
figsize=(18, 14) figsize=(18, 14)
) )