Отливать изображение в float32

tf.cast(x=image, dtype=tf.float32)
HighKage