How to build a multi-class convolutional neural network with Keras

Question

I am trying to implement a U-Net with Keras with Tensorflow backend for an image segmentation task. I have images of size (128,96) as input to the network together with mask ima

Answer 1

I don't see your prediction layer which as far as I know must be a dense layer and not a convolutional layer. Maybe that's your problem.