I’m working on an image segmentation project in Flutter using TensorFlow Lite. My model is supposed to segment an image into 4 classes, but the output I’m getting only shows 2 classes. Here’s the relevant part of my code:

I have a flutter code to integrate with tflite. I’m sure that I’ve added my model in assests.