How to Get labelled Output

[ZiqiFrancisFan]

I am trying to get the labelled output using the trained model. The test.py only provides the RGB output. Is there an easy way to get the labelled outputs of test inputs using the trained model? Thanks a lot for your advice!