You can use tf.imageor any other library you prefer.