Getting class labels back from predictions

[wnorcbrown]

Is there any utility I can use for getting the class labels back from the predictions?

I would like to know which classes the model is predicting from the ADE20K dataset.

Thank you

[NiccoloCavagnero]

Hi @wnorcbrown,

Unfortunately there is not such utility, but there is a pull request that can be handy for you to achieve what you want.

Best,
Nick

[wnorcbrown]

Thanks @NiccoloCavagnero ,

Which pull request is that?

[NiccoloCavagnero]

Here.

Best,
Nick