I got torch.Size([1, 150, 256, 256]), since there are 150 classes in ADE20k dataset.
However, I did not find anywhere in the code, which prediction of the output[0, :, x, y] corresponds to which of the 150 classes in ADE dataset. Where can I find list of classes?
When I output with any model, I got probabilities per class on each pixel. E.g. when I run
I got
torch.Size([1, 150, 256, 256])
, since there are 150 classes in ADE20k dataset.However, I did not find anywhere in the code, which prediction of the
output[0, :, x, y]
corresponds to which of the 150 classes in ADE dataset. Where can I find list of classes?