I am trying to train a FCN model for segmentation. I have an array of train images of size (256,256,3) and another array of ground truth(masks) of size (256,256,3). My model