mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-25 10:32:59 +00:00
dlr: we can't plot the input tensor because it has 8 channels rather than 1 or 3
This commit is contained in:
parent
373dda03b5
commit
3787155665
1 changed files with 4 additions and 1 deletions
|
@ -262,7 +262,10 @@ def plot_predictions(filepath, input_items, colormap, model):
|
||||||
|
|
||||||
plot_samples_matplotlib(
|
plot_samples_matplotlib(
|
||||||
filepath,
|
filepath,
|
||||||
[input_tensor, prediction_colormap],
|
[
|
||||||
|
# input_tensor,
|
||||||
|
prediction_colormap
|
||||||
|
],
|
||||||
figsize=(18, 14)
|
figsize=(18, 14)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue