I’m trying to convert a .h5 model, which is a simplified version of Unet for image segmentation, into a .tflite model. Does anyone know the correct way to do this? So far, the .h5 model seems to make good predictions, but when I try the .tflite model, the predictions are literally an image completely in purple color.