Hello,
Right now I have pseudo-labels for an image dataset from DeepLabV3+ that are of shape [H W, 1] where each pixel is an integer from 0 to 18 representing the class of the pixel (using cityscapes pretraied models).
Now, what I want to do is focus on only a few classes (say for example, 10, 12, 13) and map the rest of the classes to some background class. What is the best way to do this in PyTorch?
I guess numpy would also work.
Thanks!