Thank you for your continued response. I tried your code snippet and it did give me a random segmented output.
As for your question
“Did you observe your training and validation loss and accuracy?”
I am not sure how I can visualize training, validation loss and accuracy during training. Also, how do I implement validation and validation loss during the training epoch?
How do I know if the model is overfitting for one of the class? Also, how do I correct for it?
Thanks
Nishanth