...tf.keras.metrics.MeanIoU support y_pred from logits...
tf.keras.metrics.MeanIoU doesn't have an option that allow y_pred passed as logits, we want to pass y_pred as a logits tensor, you can simply add an optional step into the update_state function of this class: y_