Отливать изображение в float32
tf.cast(x=image, dtype=tf.float32)
HighKage
tf.cast(x=image, dtype=tf.float32)