How to achieve multiple category segmentation

[wxnzxt]

First of all, thank you very much for your sharing. I would like to ask how to achieve multiple category segmentation. I tried to change out_channels and it can only be equal to 1. because label's channel is 1. How can I set the number of categories? I want the output class to be 18. I would be grateful if someone could answer.