mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-16 14:43:01 +00:00
dlr: tf.Tensor → numpy array → list → json → disk
This commit is contained in:
parent
b287970032
commit
5e79825292
1 changed files with 1 additions and 1 deletions
|
@ -341,7 +341,7 @@ def plot_predictions(filepath, input_items, colormap, model):
|
|||
|
||||
save_samples(
|
||||
filepath_jsonl,
|
||||
prediction_mask
|
||||
prediction_mask.numpy().list()
|
||||
)
|
||||
i += 1
|
||||
|
||||
|
|
Loading…
Reference in a new issue