I am implementing a paper of semantic segmentation task. In it they have mentioned about adding the classification head. What exactly is it and how to add in the model? Segmentation models pytorch library provides classification head but I am unable to figure it out.
“Apart from UNet CNN, we add a classification head to each architecture for up-sampling the complex low-level features, and map the high-level features to output for semantic segmentation”