I know this question already has a clear answer about how to visualize segmentation result. My question is if I'm using the CrossEntropyLoss as the loss function, after training. The predicted classification (segmentation) result doesn't represent the class 0,1,2. It represents the unnormalized Log probability as told in cs231n standford class. So if I want to visualize the segmentaion result, should I use the
to get the classes and then plot them out? or should I do a maximum function over the first dimension of my output.
For example, if the predicted output is size 3x1024x1024, how could I visualize the segmenation result?
Thank you in advance.