what is the correct way to reshape image segmentation model output from 2D (num of classes, num of pixels) to 3D channel last images
问题 I am using keras and python for satellite image segmentation. It is my understanding that to get (pixel level)predictions for image segmentation, model reshapes layer of dimension(-1,num_classes,height,width) to shape (-1,num_classes,height*width).This is then followed by applying activation function like softmax or sigmoid. My question is how to recover images after this step back in the format either channel first or channel last? example code o = (Reshape(( num_classes , outputHeight