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