What is the shape of your output activation?
In a segmentation use case, your output activation will most likely have the shape [batch_size, nb_classes, height, width].
If nb_classes was defined as 22, you will only get these labels.