import tensorflow as tf def process(image): image= tf.cast(image/255, tf.float32) return image