mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-04 17:13:02 +00:00
fixup
This commit is contained in:
parent
54a841efe9
commit
6c09d5254d
1 changed files with 1 additions and 0 deletions
|
@ -113,6 +113,7 @@ def do_png(args, ai, dataset, model_params):
|
|||
water_predict = tf.math.argmax(water_predict, axis=-1)
|
||||
# [ width, height, bins ]
|
||||
water_actual = tf.squeeze(water[i_batch])
|
||||
# [ width, height ]
|
||||
water_actual = tf.math.argmax(water_actual, axis=-1)
|
||||
|
||||
segmentation_plot(
|
||||
|
|
Loading…
Reference in a new issue